Tree: Format

Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
kingbri
2025-10-09 22:33:53 -04:00
parent 4235f98e83
commit 85459ce600
2 changed files with 7 additions and 4 deletions

View File

@@ -1,6 +1,8 @@
import platform
import torch
from loguru import logger
def exllama_supports_nccl():
if platform.system() == "Windows":
unsupported_message = (
@@ -9,5 +11,4 @@ def exllama_supports_nccl():
logger.warning(unsupported_message)
return False
import torch
return torch.cuda.is_available() and torch.distributed.is_nccl_available()