Update nodes_model_loading.py

This commit is contained in:
kijai
2025-08-15 14:45:53 +03:00
parent 0beb5a8911
commit 2df76f3363
+7
View File
@@ -751,6 +751,13 @@ class WanVideoModelLoader:
fantasytalking_model=None, multitalk_model=None, fantasyportrait_model=None):
assert not (vram_management_args is not None and block_swap_args is not None), "Can't use both block_swap_args and vram_management_args at the same time"
if torch.cuda.is_available():
#only warning for now
major, minor = torch.cuda.get_device_capability(device)
log.info(f"CUDA Compute Capability: {major}.{minor}")
if compile_args is not None and "e4" in quantization and (major, minor) < (8, 9):
log.warning("Torch.compile with fp8_e4m3fn weights on CUDA compute capability < 8.9 is not supported. Please use fp8_e5m2, GGUF or higher precision instead.")
lora_low_mem_load = merge_loras = False
if lora is not None:
for l in lora: