charliebaby2023 commited on
Commit
db7f669
·
verified ·
1 Parent(s): 9605c55

Update app_demo.py

Browse files
Files changed (1) hide show
  1. 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 = None,
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)