bryanbrunetti commited on
Commit
b555fee
·
1 Parent(s): 56a7978

fix lora path

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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], weight_name=lora_weights[lora_model]["weight_name"])
 
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