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

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
  #"SimianLuo/LCM_Dreamshaper_v7"
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
  )
 
33
  #"SimianLuo/LCM_Dreamshaper_v7"
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
  )