Hugging Face's logo

Spaces:
r3gm
/
Fast_Stable_diffusion_CPU
Running

App Files Community
SimianLuo commited on
Commit
ef9a28b
•
1 Parent(s): c144567

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "768"))
31
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
32
  DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a bit worse
33
 
34
- pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img")
35
  pipe.to(torch_device="cuda", torch_dtype=DTYPE)
36
 
37
 
 
31
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
32
  DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a bit worse
33
 
34
+ pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", revision="main")
35
  pipe.to(torch_device="cuda", torch_dtype=DTYPE)
36
 
37