prithivMLmods commited on
Commit
6628dd9
·
verified ·
1 Parent(s): 18c4ed2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -210,7 +210,7 @@ def adjust_generation_mode(speed_mode):
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=""):
@@ -257,7 +257,7 @@ def process_adapter_generation(prompt, cfg_scale, steps, selected_index, randomi
257
  pipe.unload_lora_weights()
258
 
259
  # Load LoRAs based on speed mode
260
- if speed_mode == "Speed (8 steps)":
261
  with Timer("Loading Lightning LoRA and style LoRA"):
262
  # Load Lightning LoRA first
263
  pipe.load_lora_weights(
 
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, 4.0
214
 
215
  @spaces.GPU(duration=108)
216
  def create_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, negative_prompt=""):
 
257
  pipe.unload_lora_weights()
258
 
259
  # Load LoRAs based on speed mode
260
+ if speed_mode == "Fast (8 steps)":
261
  with Timer("Loading Lightning LoRA and style LoRA"):
262
  # Load Lightning LoRA first
263
  pipe.load_lora_weights(