Spaces:
Running
on
Zero
Running
on
Zero
amazonaws-la
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -55,6 +55,7 @@ def generate(
|
|
55 |
model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
|
56 |
vaecall = 'stabilityai/sd-vae-ft-mse',
|
57 |
lora = 'amazonaws-la/juliette',
|
|
|
58 |
) -> PIL.Image.Image:
|
59 |
if torch.cuda.is_available():
|
60 |
|
@@ -144,6 +145,13 @@ with gr.Blocks(css="style.css") as demo:
|
|
144 |
model = gr.Text(label='Modelo')
|
145 |
vaecall = gr.Text(label='VAE')
|
146 |
lora = gr.Text(label='LoRA')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
147 |
with gr.Row():
|
148 |
prompt = gr.Text(
|
149 |
label="Prompt",
|
@@ -322,6 +330,7 @@ with gr.Blocks(css="style.css") as demo:
|
|
322 |
model,
|
323 |
vaecall,
|
324 |
lora,
|
|
|
325 |
],
|
326 |
outputs=result,
|
327 |
api_name="run",
|
|
|
55 |
model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
|
56 |
vaecall = 'stabilityai/sd-vae-ft-mse',
|
57 |
lora = 'amazonaws-la/juliette',
|
58 |
+
lora_scale: float = 0.7,
|
59 |
) -> PIL.Image.Image:
|
60 |
if torch.cuda.is_available():
|
61 |
|
|
|
145 |
model = gr.Text(label='Modelo')
|
146 |
vaecall = gr.Text(label='VAE')
|
147 |
lora = gr.Text(label='LoRA')
|
148 |
+
lora_scale = = gr.Slider(
|
149 |
+
label="Lora Scale",
|
150 |
+
minimum=0.01,
|
151 |
+
maximum=1,
|
152 |
+
step=0.01,
|
153 |
+
value=0.7,
|
154 |
+
)
|
155 |
with gr.Row():
|
156 |
prompt = gr.Text(
|
157 |
label="Prompt",
|
|
|
330 |
model,
|
331 |
vaecall,
|
332 |
lora,
|
333 |
+
lora_scale,
|
334 |
],
|
335 |
outputs=result,
|
336 |
api_name="run",
|