fffiloni commited on
Commit
f69042b
·
1 Parent(s): dfd27a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -102,6 +102,8 @@ def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, l
102
  generator=generator,
103
  cross_attention_kwargs={"scale": lora_weight}
104
  ).images[0]
 
 
105
 
106
  return image
107
 
@@ -180,6 +182,7 @@ with gr.Blocks(css=css) as demo:
180
  label="Safetensors file",
181
  #value="pytorch_lora_weights.safetensors",
182
  info="specify which one if model has several .safetensors files",
 
183
  visible = False
184
  )
185
  with gr.Column():
 
102
  generator=generator,
103
  cross_attention_kwargs={"scale": lora_weight}
104
  ).images[0]
105
+
106
+ pipe.unfuse_lora()
107
 
108
  return image
109
 
 
182
  label="Safetensors file",
183
  #value="pytorch_lora_weights.safetensors",
184
  info="specify which one if model has several .safetensors files",
185
+ allow_custom_value=True,
186
  visible = False
187
  )
188
  with gr.Column():