From d22c0b866bc9ab52a07d3d054a230c0efd6f545b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?= <40791699+kijai@users.noreply.github.com> Date: Fri, 20 Sep 2024 21:36:42 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 5969e24..4c06493 100644 --- a/nodes.py +++ b/nodes.py @@ -576,16 +576,16 @@ class CogVideoSampler: offload_device = mm.unet_offload_device() pipe = pipeline["pipe"] dtype = pipeline["dtype"] - + scheduler_config = pipeline["scheduler_config"] if not pipeline["cpu_offloading"]: pipe.transformer.to(device) generator = torch.Generator(device=device).manual_seed(seed) 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": - pipe.scheduler = CogVideoXDPMScheduler.from_pretrained(base_path, subfolder="scheduler") + pipe.scheduler = CogVideoXDPMScheduler.from_config(scheduler_config) if negative.shape[1] < positive.shape[1]: target_length = positive.shape[1]