Spaces:
Running
on
Zero
Running
on
Zero
arthur-qiu
commited on
Commit
•
c36e301
1
Parent(s):
eb199d6
init
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from PIL import Image
|
|
| 8 |
from pipeline_freescale import StableDiffusionXLPipeline
|
| 9 |
from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
|
| 10 |
|
| 11 |
-
@spaces.GPU
|
| 12 |
def infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu):
|
| 13 |
pipe = pipe.to("cuda")
|
| 14 |
generator = torch.Generator(device='cuda')
|
|
|
|
| 8 |
from pipeline_freescale import StableDiffusionXLPipeline
|
| 9 |
from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
|
| 10 |
|
| 11 |
+
@spaces.GPU(duration=120)
|
| 12 |
def infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu):
|
| 13 |
pipe = pipe.to("cuda")
|
| 14 |
generator = torch.Generator(device='cuda')
|