Disable comfy kitchen cuda if pytorch cuda less than 13 (#11681)

2026-01-09 05:40:49 +08:00 · 2026-01-06 19:13:43 -08:00 · 2026-01-06 19:13:43 -08:00 · edee33f55e
commit edee33f55e
parent 2c03884f5f
1 changed files with 7 additions and 0 deletions
--- a/comfy/quant_ops.py
+++ b/comfy/quant_ops.py
@ -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}")