charliebaby2023 commited on
Commit
7015c75
·
verified ·
1 Parent(s): 93ab87a

Update app_demo.py

Browse files
Files changed (1) hide show
  1. app_demo.py +4 -4
app_demo.py CHANGED
@@ -30,20 +30,20 @@ DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a
30
 
31
 
32
  #"SimianLuo/LCM_Dreamshaper_v7"
33
- pipe = DiffusionPipeline.from_pretrained( "charliebaby2023/cybrpny" , custom_pipeline="latent_consistency_txt2img",
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)
40
- pipe.to(torch_dtype="float16" )
41
  pipe.to("cpu")
42
 
43
 
44
  #from diffusers import StableDiffusionPipeline
45
 
46
- model_id = "charliebaby2023/cybrpny"
47
 
48
  #pipe = StableDiffusionPipeline.from_pretrained(model_id,)
49
  #prompt = "A futuristic cityscape at sunset"
 
30
 
31
 
32
  #"SimianLuo/LCM_Dreamshaper_v7"
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)
40
+ pipe.to(torch_dtype="float32" )
41
  pipe.to("cpu")
42
 
43
 
44
  #from diffusers import StableDiffusionPipeline
45
 
46
+ model_id = "SimianLuo/LCM_Dreamshaper_v7"
47
 
48
  #pipe = StableDiffusionPipeline.from_pretrained(model_id,)
49
  #prompt = "A futuristic cityscape at sunset"