Spaces:
Sleeping
Sleeping
Commit
·
b555fee
1
Parent(s):
56a7978
fix lora path
Browse files
app.py
CHANGED
@@ -41,7 +41,8 @@ def infer(prompt, lora_models, seed=42, randomize_seed=False, width=1024, height
|
|
41 |
try:
|
42 |
for lora_model in lora_models:
|
43 |
print(f"loading LoRA: {lora_model}")
|
44 |
-
pipe.load_lora_weights(lora_weights[lora_model]
|
|
|
45 |
except Exception as e:
|
46 |
return None, seed, f"Failed to load LoRA model: {str(e)}"
|
47 |
|
|
|
41 |
try:
|
42 |
for lora_model in lora_models:
|
43 |
print(f"loading LoRA: {lora_model}")
|
44 |
+
pipe.load_lora_weights(lora_weights[lora_model]["path"],
|
45 |
+
weight_name=lora_weights[lora_model]["weight_name"])
|
46 |
except Exception as e:
|
47 |
return None, seed, f"Failed to load LoRA model: {str(e)}"
|
48 |
|