Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse filesrun on zero gpu
app.py
CHANGED
@@ -17,7 +17,7 @@ pipe.to("cuda")
|
|
17 |
|
18 |
MAX_SEED = 2**32-1
|
19 |
|
20 |
-
|
21 |
def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
|
22 |
# Set random seed for reproducibility
|
23 |
if randomize_seed:
|
|
|
17 |
|
18 |
MAX_SEED = 2**32-1
|
19 |
|
20 |
+
@spaces.GPU(duration=80)
|
21 |
def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
|
22 |
# Set random seed for reproducibility
|
23 |
if randomize_seed:
|