From 7141ab23a5f81eb1d4c89b47082bd0b1a107ef4b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 22 Oct 2024 01:58:34 +0300
Subject: [PATCH] Update pipeline_cogvideox.py

---
 pipeline_cogvideox.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pipeline_cogvideox.py b/pipeline_cogvideox.py
index 8a2e15b..677402a 100644
--- a/pipeline_cogvideox.py
+++ b/pipeline_cogvideox.py
@@ -609,6 +609,11 @@ class CogVideoXPipeline(VideoSysPipeline):
             controlnet_states = None
             control_weights= None
 
+        if tora is not None:
+            for module in self.transformer.fuser_list:
+                for param in module.parameters():
+                    param.data = param.data.to(device)
+
         # 10. Denoising loop
         with self.progress_bar(total=num_inference_steps) as progress_bar:    
             old_pred_original_sample = None # for DPM-solver++