Skip to content

Commit aee2212

Browse files
comfyanonymouslrivera
authored andcommitted
Disable comfy kitchen cuda if pytorch cuda less than 13 (Comfy-Org#11681)
1 parent eae243d commit aee2212

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

comfy/quant_ops.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,13 @@
1313
get_layout_class,
1414
)
1515
_CK_AVAILABLE = True
16+
if torch.version.cuda is None:
17+
ck.registry.disable("cuda")
18+
else:
19+
cuda_version = tuple(map(int, str(torch.version.cuda).split('.')))
20+
if cuda_version < (13,):
21+
ck.registry.disable("cuda")
22+
1623
ck.registry.disable("triton")
1724
for k, v in ck.list_backends().items():
1825
logging.info(f"Found comfy_kitchen backend {k}: {v}")

0 commit comments

Comments
 (0)