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

Update pipeline_cogvideox.py

Browse Source
This commit is contained in:
kijai 2024-10-22 01:58:34 +03:00
parent 35cb4fde94
commit 7141ab23a5
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pipeline_cogvideox.py
View File

@ -609,6 +609,11 @@ class CogVideoXPipeline(VideoSysPipeline):
controlnet_states = None controlnet_states = None
control_weights= 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 # 10. Denoising loop
with self.progress_bar(total=num_inference_steps) as progress_bar: with self.progress_bar(total=num_inference_steps) as progress_bar:
old_pred_original_sample = None # for DPM-solver++ old_pred_original_sample = None # for DPM-solver++