Commit
•
331f30a
1
Parent(s):
29f3401
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ pipe = StableDiffusionXLPipeline(
|
|
| 32 |
pipe = pipe.to("cuda")
|
| 33 |
pipe.enable_model_cpu_offload()
|
| 34 |
|
| 35 |
-
@spaces.GPU
|
| 36 |
def generate_image(prompt, negative_prompt, height, width, num_inference_steps, guidance_scale, num_images_per_prompt, use_random_seed, seed, progress=gr.Progress(track_tqdm=True)):
|
| 37 |
if use_random_seed:
|
| 38 |
seed = random.randint(0, 2**32 - 1)
|
|
|
|
| 32 |
pipe = pipe.to("cuda")
|
| 33 |
pipe.enable_model_cpu_offload()
|
| 34 |
|
| 35 |
+
@spaces.GPU(duration=200)
|
| 36 |
def generate_image(prompt, negative_prompt, height, width, num_inference_steps, guidance_scale, num_images_per_prompt, use_random_seed, seed, progress=gr.Progress(track_tqdm=True)):
|
| 37 |
if use_random_seed:
|
| 38 |
seed = random.randint(0, 2**32 - 1)
|