Update app_demo.py
Browse files- 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 |
-
|
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)
|