diff --git a/nodes.py b/nodes.py index fc853c2..4577310 100644 --- a/nodes.py +++ b/nodes.py @@ -737,7 +737,8 @@ class CogVideoXFunSampler: base_path = pipeline["base_path"] assert "Fun" in base_path, "'Unfun' models not supported in 'CogVideoXFunSampler', use the 'CogVideoSampler'" - pipe.enable_model_cpu_offload(device=device) + if not pipeline["cpu_offloading"]: + pipe.enable_model_cpu_offload(device=device) mm.soft_empty_cache()