rupeshs commited on
Commit
5993d69
1 Parent(s): e954e95
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +2 -2
frontend/webui/hf_demo.py CHANGED
@@ -55,7 +55,7 @@ def predict(
55
  # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
56
  lcm_diffusion_setting.image_width = 512
57
  lcm_diffusion_setting.image_height = 512
58
- lcm_diffusion_setting.use_openvino = True
59
  lcm_diffusion_setting.use_tiny_auto_encoder = True
60
  pprint(lcm_diffusion_setting.model_dump())
61
  lcm_text_to_image.init(lcm_diffusion_setting=lcm_diffusion_setting)
@@ -101,7 +101,7 @@ def _get_footer_message() -> str:
101
 
102
  with gr.Blocks(css=css) as demo:
103
  with gr.Column(elem_id="container"):
104
- use_openvino = "- OpenVINO" if is_openvino_device() else ""
105
  gr.Markdown(
106
  f"""# FastSD CPU demo {use_openvino}
107
  **Device : {DEVICE.upper()} , {get_device_name()}**
 
55
  # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
56
  lcm_diffusion_setting.image_width = 512
57
  lcm_diffusion_setting.image_height = 512
58
+ lcm_diffusion_setting.use_openvino = False
59
  lcm_diffusion_setting.use_tiny_auto_encoder = True
60
  pprint(lcm_diffusion_setting.model_dump())
61
  lcm_text_to_image.init(lcm_diffusion_setting=lcm_diffusion_setting)
 
101
 
102
  with gr.Blocks(css=css) as demo:
103
  with gr.Column(elem_id="container"):
104
+ use_openvino = "" if is_openvino_device() else ""
105
  gr.Markdown(
106
  f"""# FastSD CPU demo {use_openvino}
107
  **Device : {DEVICE.upper()} , {get_device_name()}**