Hugging Face's logo

Spaces:
wangfuyun
/
Phased-Consistency-Model-PCM
Running on Zero

App Files Community
2
wangfuyun commited on
Commit
89883a7
·
verified ·
1 Parent(s): ad4d334

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ if torch.cuda.is_available():
34
  variant="fp16",
35
  ).to("cuda")
36
  pipe_sd15 = StableDiffusionPipeline.from_pretrained(
37
- "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
38
  ).to("cuda")
39
 
40
  if SAFETY_CHECKER:
 
34
  variant="fp16",
35
  ).to("cuda")
36
  pipe_sd15 = StableDiffusionPipeline.from_pretrained(
37
+ "ZeroCool94/stable-diffusion-v1-5", torch_dtype=torch.float16,
38
  ).to("cuda")
39
 
40
  if SAFETY_CHECKER: