charliebaby2023 commited on
Commit
4f47622
·
verified ·
1 Parent(s): 7015c75

Update app_demo.py

Browse files
Files changed (1) hide show
  1. app_demo.py +1 -1
app_demo.py CHANGED
@@ -33,7 +33,7 @@ DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a
33
  pipe = DiffusionPipeline.from_pretrained( "SimianLuo/LCM_Dreamshaper_v7" , custom_pipeline="SimianLuo/LCM_Dreamshaper_v7",
34
  custom_revision="main",
35
  low_cpu_mem_usage=True,
36
- pipe.safety_checker= None,
37
  use_safetensors=True
38
  )
39
  #pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)
 
33
  pipe = DiffusionPipeline.from_pretrained( "SimianLuo/LCM_Dreamshaper_v7" , custom_pipeline="SimianLuo/LCM_Dreamshaper_v7",
34
  custom_revision="main",
35
  low_cpu_mem_usage=True,
36
+ safety_checker= None,
37
  use_safetensors=True
38
  )
39
  #pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)