Hugging Face's logo

Spaces:
cocktailpeanut
/
Phased-Consistency-Model-PCM
Runtime error

App Files Community
cocktailpeanut commited on
Commit
ce4b9eb
•
1 Parent(s): 49901fa

update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import devicetorch
4
  from diffusers import StableDiffusionXLPipeline, StableDiffusionPipeline, LCMScheduler
5
  from diffusers.schedulers import TCDScheduler
6
 
7
- import spaces
8
  from PIL import Image
9
 
10
  checkpoints = {
@@ -46,7 +46,7 @@ pipe_sd15 = StableDiffusionPipeline.from_pretrained(
46
  "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
47
  ).to(device)
48
 
49
- @spaces.GPU(enable_queue=True)
50
  def generate_image(
51
  prompt,
52
  ckpt,
 
4
  from diffusers import StableDiffusionXLPipeline, StableDiffusionPipeline, LCMScheduler
5
  from diffusers.schedulers import TCDScheduler
6
 
7
+ #import spaces
8
  from PIL import Image
9
 
10
  checkpoints = {
 
46
  "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
47
  ).to(device)
48
 
49
+ #@spaces.GPU(enable_queue=True)
50
  def generate_image(
51
  prompt,
52
  ckpt,