jiuface commited on
Commit
e91fd6d
·
verified ·
1 Parent(s): a209f33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -167,7 +167,8 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, s
167
  with calculateDuration("Unloading LoRA"):
168
  img2img_pipe.unload_lora_weights()
169
  txt2img_pipe.unload_lora_weights()
170
-
 
171
  lora_configs = None
172
  adapter_names = []
173
  lora_names = []
@@ -187,7 +188,8 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, s
187
  lora_repo = lora_info.get("repo")
188
  weights = lora_info.get("weights")
189
  adapter_name = lora_info.get("adapter_name")
190
- lora_name = f"lora_{generate_random_4_digit_string()}"
 
191
  lora_names.append(load_image)
192
  adapter_weight = lora_info.get("adapter_weight")
193
  adapter_names.append(adapter_name)
@@ -202,7 +204,7 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, s
202
  print("load lora error")
203
 
204
  # set lora weights
205
- if len(adapter_names) > 0:
206
  if img2img_model:
207
  img2img_pipe.set_adapters(lora_names, adapter_weights=adapter_weights)
208
  else:
 
167
  with calculateDuration("Unloading LoRA"):
168
  img2img_pipe.unload_lora_weights()
169
  txt2img_pipe.unload_lora_weights()
170
+ print(pipe.get_active_adapters())
171
+
172
  lora_configs = None
173
  adapter_names = []
174
  lora_names = []
 
188
  lora_repo = lora_info.get("repo")
189
  weights = lora_info.get("weights")
190
  adapter_name = lora_info.get("adapter_name")
191
+
192
+ lora_name = generate_random_4_digit_string()
193
  lora_names.append(load_image)
194
  adapter_weight = lora_info.get("adapter_weight")
195
  adapter_names.append(adapter_name)
 
204
  print("load lora error")
205
 
206
  # set lora weights
207
+ if len(lora_names) > 0:
208
  if img2img_model:
209
  img2img_pipe.set_adapters(lora_names, adapter_weights=adapter_weights)
210
  else: