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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-09-18 21:09:49 +03:00
parent 251d6599c0
commit 1e15c5b901
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -482,7 +482,7 @@ class CogVideoXFunSampler:
pipe = pipeline["pipe"]
dtype = pipeline["dtype"]
pipe.enable_model_cpu_offload()
pipe.enable_model_cpu_offload(device=device)
mm.soft_empty_cache()
@ -594,7 +594,7 @@ class CogVideoXFunVid2VidSampler:
pipe = pipeline["pipe"]
dtype = pipeline["dtype"]
pipe.enable_model_cpu_offload()
pipe.enable_model_cpu_offload(device=device)
mm.soft_empty_cache()