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

Update nodes.py

Browse Source
This commit is contained in:
Jukka Seppänen 2024-09-20 21:36:42 +03:00
parent baabbf9a46
commit d22c0b866b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -576,16 +576,16 @@ class CogVideoSampler:
offload_device = mm.unet_offload_device() offload_device = mm.unet_offload_device()
pipe = pipeline["pipe"] pipe = pipeline["pipe"]
dtype = pipeline["dtype"] dtype = pipeline["dtype"]
scheduler_config = pipeline["scheduler_config"]
if not pipeline["cpu_offloading"]: if not pipeline["cpu_offloading"]:
pipe.transformer.to(device) pipe.transformer.to(device)
generator = torch.Generator(device=device).manual_seed(seed) generator = torch.Generator(device=device).manual_seed(seed)
if scheduler == "DDIM" or scheduler == "DDIM_tiled": if scheduler == "DDIM" or scheduler == "DDIM_tiled":
pipe.scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder="scheduler") pipe.scheduler = CogVideoXDDIMScheduler.from_config(scheduler_config)
elif scheduler == "DPM": elif scheduler == "DPM":
pipe.scheduler = CogVideoXDPMScheduler.from_pretrained(base_path, subfolder="scheduler") pipe.scheduler = CogVideoXDPMScheduler.from_config(scheduler_config)
if negative.shape[1] < positive.shape[1]: if negative.shape[1] < positive.shape[1]:
target_length = positive.shape[1] target_length = positive.shape[1]