Hugging Face's logo

Spaces:
MoonQiu
/
FreeScale
Running on Zero

App Files Community
2
Anonymous commited on
Commit
c4f1e79
·
1 Parent(s): 1b8902c

load in advance

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -15,7 +15,6 @@ model_ckpt = "stabilityai/stable-diffusion-xl-base-1.0"
15
  model_ckpt_turbo = "stabilityai/sdxl-turbo"
16
  pipe = StableDiffusionXLPipeline.from_pretrained(model_ckpt, torch_dtype=dtype).to(device)
17
  pipe_turbo = StableDiffusionXLPipeline_Turbo.from_pretrained(model_ckpt_turbo, torch_dtype=dtype).to(device)
18
- generator = torch.Generator(device='cuda')
19
  torch.cuda.empty_cache()
20
 
21
  @spaces.GPU(duration=120)
 
15
  model_ckpt_turbo = "stabilityai/sdxl-turbo"
16
  pipe = StableDiffusionXLPipeline.from_pretrained(model_ckpt, torch_dtype=dtype).to(device)
17
  pipe_turbo = StableDiffusionXLPipeline_Turbo.from_pretrained(model_ckpt_turbo, torch_dtype=dtype).to(device)
 
18
  torch.cuda.empty_cache()
19
 
20
  @spaces.GPU(duration=120)