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