charliebaby2023 commited on
Commit
b9c41a1
·
verified ·
1 Parent(s): 498a837

Update app_demo.py

Browse files
Files changed (1) hide show
  1. app_demo.py +1 -1
app_demo.py CHANGED
@@ -37,7 +37,7 @@ model_cache = {}
37
 
38
  #custom
39
  model_id = "Lykon/dreamshaper-xl-v2-turbo"
40
- custom_pipe = DiffusionPipeline.from_pretrained(mode_id, custom_pipeline="latent_consistency_txt2img", custom_revision="main")
41
  #1st
42
  pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
43
  pipe.to(torch_device="cpu", torch_dtype=DTYPE)
 
37
 
38
  #custom
39
  model_id = "Lykon/dreamshaper-xl-v2-turbo"
40
+ custom_pipe = DiffusionPipeline.from_pretrained(model_id, custom_pipeline="latent_consistency_txt2img", custom_revision="main")
41
  #1st
42
  pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
43
  pipe.to(torch_device="cpu", torch_dtype=DTYPE)