Hugging Face's logo

Spaces:
mohan32189
/
Fast_Stable_diffusion_CPU
Runtime error

App Files Community
r3gm commited on
Commit
20c3d75
·
1 Parent(s): 5f3bf02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
36
  DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a bit worse
37
 
38
  pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
39
- pipe.to(torch_device="cuda", torch_dtype=DTYPE)
40
 
41
 
42
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
36
  DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a bit worse
37
 
38
  pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
39
+ pipe.to(torch_device="cpu", torch_dtype=DTYPE)
40
 
41
 
42
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: