From f7a88cbd56316378e48bdc7383c0afd5d87125d4 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 8 Nov 2024 21:23:29 +0200
Subject: [PATCH] Update model_loading.py

---
 model_loading.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model_loading.py b/model_loading.py
index b8bb57a..03513cf 100644
--- a/model_loading.py
+++ b/model_loading.py
@@ -4,6 +4,8 @@ import torch.nn as nn
 import json
 import folder_paths
 import comfy.model_management as mm
+from .utils import check_diffusers_version, remove_specific_blocks, log
+check_diffusers_version()
 
 from diffusers.models import AutoencoderKLCogVideoX
 from diffusers.schedulers import CogVideoXDDIMScheduler
@@ -20,7 +22,6 @@ from .cogvideox_fun.pipeline_cogvideox_control import CogVideoX_Fun_Pipeline_Con
 
 from .videosys.cogvideox_transformer_3d import CogVideoXTransformer3DModel as CogVideoXTransformer3DModelPAB
 
-from .utils import check_diffusers_version, remove_specific_blocks, log
 from comfy.utils import load_torch_file
 
 script_directory = os.path.dirname(os.path.abspath(__file__))
@@ -311,7 +312,6 @@ class DownloadAndLoadCogVideoGGUFModel:
                 "pab_config": ("PAB_CONFIG", {"default": None}),
                 "block_edit": ("TRANSFORMERBLOCKS", {"default": None}),
                 "compile": (["disabled","torch"], {"tooltip": "compile the model for faster inference, these are advanced options only available on Linux, see readme for more info"}),
-              
             }
         }