From 1e15c5b901b48c96b5d461a00a877cfdb3f0d99f Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 18 Sep 2024 21:09:49 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index fa88214..e1b90ba 100644 --- a/nodes.py +++ b/nodes.py @@ -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()