Update frontend/webui/hf_demo.py

frontend/webui/hf_demo.py

@@ -17,7 +17,7 @@ from transformers import pipeline
 
 lcm_text_to_image = LCMTextToImage()
 lcm_lora = LCMLora(
-    base_model_id="Lykon/dreamshaper-7",
+    base_model_id="Lykon/DreamShaper",
     lcm_lora_id="latent-consistency/lcm-lora-sdv1-5",
 )
 classifier = pipeline(
@@ -60,8 +60,8 @@ def predict(
     lcm_diffusion_setting.use_tiny_auto_encoder = False
     # lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
     # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
-    lcm_diffusion_setting.image_width = 800
-    lcm_diffusion_setting.image_height = 1600
+    lcm_diffusion_setting.image_width = 512
+    lcm_diffusion_setting.image_height = 512
     lcm_diffusion_setting.use_openvino = True
     lcm_diffusion_setting.use_tiny_auto_encoder = True
     pprint(lcm_diffusion_setting.model_dump())
@@ -150,7 +150,7 @@ with gr.Blocks(css=css) as demo:
                 label="Steps",
                 value=1,
                 minimum=1,
-                maximum=50,
+                maximum=25,
                 step=1,
             )
             seed = gr.Slider(