jiuface commited on
Commit
41a1971
·
verified ·
1 Parent(s): 1241278

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -146,6 +146,8 @@ def run_lora(prompt, lora_strings_json, cfg_scale, steps, randomize_seed, seed,
146
  # set lora weights
147
  if len(adapter_names) > 0:
148
  pipe.set_adapters(adapter_names, adapter_weights=adapter_weights)
 
 
149
 
150
  # Generate image
151
  error_message = ""
 
146
  # set lora weights
147
  if len(adapter_names) > 0:
148
  pipe.set_adapters(adapter_names, adapter_weights=adapter_weights)
149
+ # mark lora as used
150
+ pipe.fuse_lora(adapter_names=adapter_names)
151
 
152
  # Generate image
153
  error_message = ""