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

This commit is contained in:
comfyanonymous 2026-01-06 19:13:43 -08:00 committed by GitHub
parent 2c03884f5f
commit edee33f55e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions

View File

@ -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}")