Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|
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 == "
|
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(
|