amazonaws-la commited on
Commit
52ca30c
·
verified ·
1 Parent(s): b04a0da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
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",