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():
|