xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-09 04:44:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix for sequential_cpu_offload in "fun" pipeline

Browse Source
This commit is contained in:
kijai 2024-09-21 18:43:26 +03:00
parent 006507a9fe
commit bc31982707
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cogvideox_fun/pipeline_cogvideox_inpaint.py
View File

@ -209,7 +209,7 @@ class CogVideoX_Fun_Pipeline_Inpaint(DiffusionPipeline):
"""
_optional_components = []
model_cpu_offload_seq = "text_encoder->vae->transformer->vae"
model_cpu_offload_seq = ">vae->transformer->vae"
_callback_tensor_inputs = [
"latents",
@ -631,7 +631,7 @@ class CogVideoX_Fun_Pipeline_Inpaint(DiffusionPipeline):
device = self._execution_device
self.vae.to(device)
#self.vae.to(device)
# here `guidance_scale` is defined analog to the guidance weight `w` of equation (2)
# of the Imagen paper: https://arxiv.org/pdf/2205.11487.pdf . `guidance_scale = 1`