charliebaby2023 commited on
Commit
108707c
·
verified ·
1 Parent(s): 5d8cfc1

Update app_demo.py

Browse files
Files changed (1) hide show
  1. app_demo.py +2 -2
app_demo.py CHANGED
@@ -31,7 +31,7 @@ DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a
31
 
32
 
33
  #"SimianLuo/LCM_Dreamshaper_v7"
34
- pipe = DiffusionPipeline.from_pretrained( "charliebaby2023/p_y" , custom_pipeline="latent_consistency_txt2img",
35
  custom_revision="main",
36
  low_cpu_mem_usage=True
37
  pipe.safety_checker = None,
@@ -44,7 +44,7 @@ pipe.to("cpu")
44
 
45
  #from diffusers import StableDiffusionPipeline
46
 
47
- model_id = "charliebaby2023/p_y"
48
 
49
  #pipe = StableDiffusionPipeline.from_pretrained(model_id,)
50
  #prompt = "A futuristic cityscape at sunset"
 
31
 
32
 
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,
 
44
 
45
  #from diffusers import StableDiffusionPipeline
46
 
47
+ model_id = "charliebaby2023/cybrpny"
48
 
49
  #pipe = StableDiffusionPipeline.from_pretrained(model_id,)
50
  #prompt = "A futuristic cityscape at sunset"