From 588294e248117105fdad02ad24f85985a531f4cc Mon Sep 17 00:00:00 2001
From: ly19965 <35597221+ly19965@users.noreply.github.com>
Date: Tue, 19 Dec 2023 16:28:02 +0800
Subject: [PATCH] fix lora load issue (#452)

---
 app.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 74292858..9585f033 100644
--- a/app.py
+++ b/app.py
@@ -250,8 +250,9 @@ def launch_pipeline(uuid,
     set_spawn_method()
     # Check character LoRA
     tmp_character_model = base_models[base_model_index]['model_id']
-    if tmp_character_model != character_model:
-        tmp_character_model = 'AI-ModelScope/stable-diffusion-xl-base-1.0'
+    if tmp_character_model != 'AI-ModelScope/stable-diffusion-xl-base-1.0':
+        tmp_character_model = character_model
+
     folder_path = join_worker_data_dir(uuid, tmp_character_model)
     folder_list = []
     if os.path.exists(folder_path):