From 28ebd9081171d85a2fdcbb415d501fb876f70f1f Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Wed, 7 Aug 2024 17:34:57 +0800
Subject: [PATCH] Update Storydiffusion_node.py

---
 Storydiffusion_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Storydiffusion_node.py b/Storydiffusion_node.py
index 3251177..8fee75d 100644
--- a/Storydiffusion_node.py
+++ b/Storydiffusion_node.py
@@ -1531,7 +1531,7 @@ class Storydiffusion_Model_Loader:
             if not use_flux:
                 vae_id = folder_paths.get_full_path("vae", vae_id)
                 pipe.vae=AutoencoderKL.from_single_file(vae_id, torch_dtype=torch.float16)
-        if not use_kolor or not use_flux:
+        if not use_kolor and not use_flux:
             pipe.scheduler = scheduler_choice.from_config(pipe.scheduler.config)
             pipe.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
         if not use_flux: