From 9a797229f251607e77951c940c9826ea29dad656 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 9 Nov 2024 15:41:57 +0200 Subject: [PATCH] check --- nodes.py | 4 +++- requirements.txt | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index 44e1d7b..4603799 100644 --- a/nodes.py +++ b/nodes.py @@ -5,6 +5,8 @@ import comfy.model_management as mm from einops import rearrange from contextlib import nullcontext +from .utils import log, check_diffusers_version +check_diffusers_version() from diffusers.schedulers import ( CogVideoXDDIMScheduler, CogVideoXDPMScheduler, @@ -42,7 +44,7 @@ from PIL import Image import numpy as np import json -from .utils import log, check_diffusers_version + script_directory = os.path.dirname(os.path.abspath(__file__)) diff --git a/requirements.txt b/requirements.txt index 8dddf7a..2b24b6a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ huggingface_hub -diffusers>=0.30.3 +diffusers>=0.31.0 accelerate>=0.33.0 einops peft \ No newline at end of file