From 0c74c201822f0430e82b1d97b32e030a9944baae Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sun, 11 Aug 2024 09:34:13 +0800
Subject: [PATCH] Update Storydiffusion_node.py

diffusers <0.30
---
 Storydiffusion_node.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/Storydiffusion_node.py b/Storydiffusion_node.py
index 0941601..4afedf5 100644
--- a/Storydiffusion_node.py
+++ b/Storydiffusion_node.py
@@ -40,11 +40,6 @@ from diffusers import (StableDiffusionXLPipeline, DiffusionPipeline, DDIMSchedul
 
 from transformers import CLIPVisionModelWithProjection
 from transformers import CLIPImageProcessor
-from .kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256 import StableDiffusionXLPipeline as StableDiffusionXLPipelineKolors
-from .kolors.models.modeling_chatglm import ChatGLMModel
-from .kolors.models.tokenization_chatglm import ChatGLMTokenizer
-from .kolors.models.unet_2d_condition import UNet2DConditionModel as UNet2DConditionModelkolor
-from .kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter import StableDiffusionXLPipeline as StableDiffusionXLPipelinekoloripadapter
 from .msdiffusion.models.projection import Resampler
 from .msdiffusion.models.model import MSAdapter
 from .msdiffusion.utils import get_phrase_idx, get_eot_idx
@@ -1436,6 +1431,11 @@ class Storydiffusion_Model_Loader:
                 )
                 set_attention_processor(pipe.unet, id_length, is_ipadapter=False)
             elif use_kolor:
+                from .kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256 import StableDiffusionXLPipeline as StableDiffusionXLPipelineKolors
+                from .kolors.models.modeling_chatglm import ChatGLMModel
+                from .kolors.models.tokenization_chatglm import ChatGLMTokenizer
+                from .kolors.models.unet_2d_condition import UNet2DConditionModel as UNet2DConditionModelkolor
+                from .kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter import StableDiffusionXLPipeline as StableDiffusionXLPipelinekoloripadapter
                 text_encoder = ChatGLMModel.from_pretrained(
                     f'{repo_id}/text_encoder',torch_dtype=torch.float16).half()
                 vae = AutoencoderKL.from_pretrained(f"{repo_id}/vae", revision=None).half()