Spaces:
Running
on
Zero
Running
on
Zero
arthur-qiu
commited on
Commit
•
e2e8155
1
Parent(s):
c36e301
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
|
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')
|