Update app_demo.py

app_demo.py

@@ -27,13 +27,19 @@ executor = ThreadPoolExecutor()
 model_cache = {}
 
 model_id = "Lykon/dreamshaper-xl-v2-turbo"
-custom_pipe = DiffusionPipeline.from_pretrained(model_id, custom_pipeline="latent_consistency_txt2img", custom_revision="main")
+#custom_pipe = DiffusionPipeline.from_pretrained(model_id, custom_pipeline="latent_consistency_txt2img", custom_revision="main")
 pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
 pipe.to(torch_device="cpu", torch_dtype=DTYPE)
 pipe.safety_checker = None
-pipe = StableDiffusionPipeline.from_pretrained(
-    model_id, safety_checker=None, torch_dtype=DTYPE, use_safetensors=True).to("cpu")
-
+#pipe = StableDiffusionPipeline.from_pretrained(    model_id, safety_checker=None, torch_dtype=DTYPE, use_safetensors=True).to("cpu")
+
+custom_pipe = DiffusionPipeline.from_pretrained(
+    model_id,
+    custom_pipeline="latent_consistency_txt2img",
+    custom_revision="main",
+    safety_checker=None,
+    feature_extractor=None
+)
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return random.randint(0, MAX_SEED) if randomize_seed else seed