Spaces:
Running
on
Zero
Running
on
Zero
Anonymous
commited on
Commit
·
1b8902c
1
Parent(s):
4d8af09
load in advance
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ pipe_turbo = StableDiffusionXLPipeline_Turbo.from_pretrained(model_ckpt_turbo, t
|
|
| 18 |
generator = torch.Generator(device='cuda')
|
| 19 |
torch.cuda.empty_cache()
|
| 20 |
|
| 21 |
-
@spaces.GPU(duration=
|
| 22 |
def infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps):
|
| 23 |
generator = torch.Generator(device='cuda')
|
| 24 |
generator = generator.manual_seed(seed)
|
|
@@ -32,7 +32,7 @@ def infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_sca
|
|
| 32 |
).images[0]
|
| 33 |
return result
|
| 34 |
|
| 35 |
-
@spaces.GPU(duration=
|
| 36 |
def infer_gpu_part_turbo(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps):
|
| 37 |
generator = torch.Generator(device='cuda')
|
| 38 |
generator = generator.manual_seed(seed)
|
|
|
|
| 18 |
generator = torch.Generator(device='cuda')
|
| 19 |
torch.cuda.empty_cache()
|
| 20 |
|
| 21 |
+
@spaces.GPU(duration=120)
|
| 22 |
def infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps):
|
| 23 |
generator = torch.Generator(device='cuda')
|
| 24 |
generator = generator.manual_seed(seed)
|
|
|
|
| 32 |
).images[0]
|
| 33 |
return result
|
| 34 |
|
| 35 |
+
@spaces.GPU(duration=40)
|
| 36 |
def infer_gpu_part_turbo(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu, restart_steps):
|
| 37 |
generator = torch.Generator(device='cuda')
|
| 38 |
generator = generator.manual_seed(seed)
|