mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-02 06:29:52 +00:00
Disable comfy kitchen cuda if pytorch cuda less than 13 (#11681)
This commit is contained in:
@@ -13,6 +13,13 @@ try:
|
||||
get_layout_class,
|
||||
)
|
||||
_CK_AVAILABLE = True
|
||||
if torch.version.cuda is None:
|
||||
ck.registry.disable("cuda")
|
||||
else:
|
||||
cuda_version = tuple(map(int, str(torch.version.cuda).split('.')))
|
||||
if cuda_version < (13,):
|
||||
ck.registry.disable("cuda")
|
||||
|
||||
ck.registry.disable("triton")
|
||||
for k, v in ck.list_backends().items():
|
||||
logging.info(f"Found comfy_kitchen backend {k}: {v}")
|
||||
|
||||
Reference in New Issue
Block a user