From 006507a9fe56154438c60b45241cb404e66b281d Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 21 Sep 2024 16:46:59 +0300
Subject: [PATCH] Don't overwrite sequential cpu offloading in the fun sampler

---
 nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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()