multimodalart HF Staff commited on
Commit
a8ffa44
·
verified ·
1 Parent(s): 7c7b227

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -231,8 +231,10 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, asp
231
  pipe.load_lora_weights(
232
  lora_path,
233
  weight_name=weight_name,
234
- low_cpu_mem_usage=True
 
235
  )
 
236
 
237
  # Set random seed for reproducibility
238
  with calculateDuration("Randomizing seed"):
 
231
  pipe.load_lora_weights(
232
  lora_path,
233
  weight_name=weight_name,
234
+ low_cpu_mem_usage=True,
235
+ adapter_name="style"
236
  )
237
+ pipe.set_adapters(["style"], adapter_weights=[lora_scale])
238
 
239
  # Set random seed for reproducibility
240
  with calculateDuration("Randomizing seed"):