Update app_demo.py
Browse files- 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(
|
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)
|