prithivMLmods commited on
Commit
18c4ed2
·
verified ·
1 Parent(s): 9e17aa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -208,9 +208,9 @@ def handle_lora_selection(evt: gr.SelectData, aspect_ratio):
208
  def adjust_generation_mode(speed_mode):
209
  """Update UI based on speed/quality toggle."""
210
  if speed_mode == "Speed (8 steps)":
211
- return gr.update(value="Speed mode selected - 8 steps with Lightning LoRA"), 8, 1.0
212
  else:
213
- return gr.update(value="Quality mode selected - 48 steps for best quality"), 48, 3.5
214
 
215
  @spaces.GPU(duration=108)
216
  def create_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, negative_prompt=""):
@@ -477,8 +477,8 @@ with gr.Blocks(theme="bethecloud/storj_theme", css=css, delete_cache=(120, 120))
477
  with gr.Row():
478
  speed_mode = gr.Dropdown(
479
  label="Generation Mode",
480
- choices=["Speed (8 steps)", "Quality (45 steps)"],
481
- value="Quality (45 steps)",
482
  )
483
 
484
  speed_status = gr.Markdown("Quality mode active", elem_id="speed_status")
 
208
  def adjust_generation_mode(speed_mode):
209
  """Update UI based on speed/quality toggle."""
210
  if speed_mode == "Speed (8 steps)":
211
+ return gr.update(value="Fast mode selected - 8 steps with Lightning LoRA"), 8, 1.0
212
  else:
213
+ return gr.update(value="Base mode selected - 48 steps for best quality"), 48, 3.5
214
 
215
  @spaces.GPU(duration=108)
216
  def create_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, negative_prompt=""):
 
477
  with gr.Row():
478
  speed_mode = gr.Dropdown(
479
  label="Generation Mode",
480
+ choices=["Fast (8 steps)", "Base (48 steps)"],
481
+ value="Base (48 steps)",
482
  )
483
 
484
  speed_status = gr.Markdown("Quality mode active", elem_id="speed_status")