SimianLuo commited on
Commit
4bbf923
1 Parent(s): 5f3bf02

update to diffuers pipeline

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,7 +35,8 @@ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "768"))
35
  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
 
 
35
  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")
39
+ # pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
40
  pipe.to(torch_device="cuda", torch_dtype=DTYPE)
41
 
42