Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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():
|