xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-23 03:44:26 +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 = [] _optional_components = []
model_cpu_offload_seq = "text_encoder->vae->transformer->vae" model_cpu_offload_seq = ">vae->transformer->vae"
_callback_tensor_inputs = [ _callback_tensor_inputs = [
"latents", "latents",
@ -631,7 +631,7 @@ class CogVideoX_Fun_Pipeline_Inpaint(DiffusionPipeline):
device = self._execution_device 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) # 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` # of the Imagen paper: https://arxiv.org/pdf/2205.11487.pdf . `guidance_scale = 1`