diff --git a/nodes_model_loading.py b/nodes_model_loading.py index e4bfffd..7a2158d 100644 --- a/nodes_model_loading.py +++ b/nodes_model_loading.py @@ -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: