Update app_demo.py
Browse files- app_demo.py +1 -1
app_demo.py
CHANGED
@@ -34,7 +34,7 @@ DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a
|
|
34 |
pipe = DiffusionPipeline.from_pretrained( "charliebaby2023/cybrpny" , custom_pipeline="latent_consistency_txt2img",
|
35 |
custom_revision="main",
|
36 |
low_cpu_mem_usage=True,
|
37 |
-
pipe.safety_checker
|
38 |
use_safetensors=True
|
39 |
)
|
40 |
#pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)
|
|
|
34 |
pipe = DiffusionPipeline.from_pretrained( "charliebaby2023/cybrpny" , custom_pipeline="latent_consistency_txt2img",
|
35 |
custom_revision="main",
|
36 |
low_cpu_mem_usage=True,
|
37 |
+
#pipe.safety_checker= None,
|
38 |
use_safetensors=True
|
39 |
)
|
40 |
#pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)
|