xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-23 02:49:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't overwrite sequential cpu offloading in the fun sampler

Browse Source
This commit is contained in:
kijai 2024-09-21 16:46:59 +03:00
parent 73fa4be48f
commit 006507a9fe
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -737,7 +737,8 @@ class CogVideoXFunSampler:
base_path = pipeline["base_path"] base_path = pipeline["base_path"]
assert "Fun" in base_path, "'Unfun' models not supported in 'CogVideoXFunSampler', use the 'CogVideoSampler'" 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() mm.soft_empty_cache()