From d0b28a87df22dc7820e1e443739d10ea34da653d Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 18 Sep 2024 19:03:11 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index a4dec80..a48c709 100644 --- a/nodes.py +++ b/nodes.py @@ -2,7 +2,7 @@ import os import torch import folder_paths import comfy.model_management as mm -from comfy.utils import ProgressBar +from comfy.utils import ProgressBar, load_torch_file from diffusers.schedulers import CogVideoXDDIMScheduler, CogVideoXDPMScheduler, DDIMScheduler, PNDMScheduler, DPMSolverMultistepScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler from diffusers.models import AutoencoderKLCogVideoX, CogVideoXTransformer3DModel @@ -504,7 +504,7 @@ class CogVideoXFunSampler: elif scheduler == "PNDM": noise_scheduler = PNDMScheduler.from_pretrained(base_path, subfolder= 'scheduler') elif scheduler == "DDIM": - noise_scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler') + noise_scheduler = DDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler') elif scheduler == "CogVideoXDDIM": noise_scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler') elif scheduler == "CogVideoXDPMScheduler":