fix lora load issue (#452)

modelscope · Dec 19, 2023 · 588294e · 588294e
1 parent d4e1693
commit 588294e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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):