rupeshs commited on
Commit
409b7dd
1 Parent(s): 673c7f1

model update

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +1 -7
frontend/webui/hf_demo.py CHANGED
@@ -37,13 +37,6 @@ def predict(
37
  seed,
38
  use_seed,
39
  ):
40
- lcm_text_to_image.init(
41
- model_id=LCM_DEFAULT_MODEL_OPENVINO,
42
- use_openvino=True,
43
- use_lora=False,
44
- lcm_lora=lcm_lora,
45
- use_tiny_auto_encoder=False,
46
- )
47
  print(f"prompt - {prompt}")
48
  lcm_diffusion_setting = LCMDiffusionSetting()
49
  lcm_diffusion_setting.prompt = prompt
@@ -56,6 +49,7 @@ def predict(
56
  lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
57
  lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
58
  lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
 
59
  start = perf_counter()
60
  images = lcm_text_to_image.generate(lcm_diffusion_setting)
61
  latency = perf_counter() - start
 
37
  seed,
38
  use_seed,
39
  ):
 
 
 
 
 
 
 
40
  print(f"prompt - {prompt}")
41
  lcm_diffusion_setting = LCMDiffusionSetting()
42
  lcm_diffusion_setting.prompt = prompt
 
49
  lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
50
  lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
51
  lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
52
+ lcm_text_to_image.init(lcm_diffusion_setting)
53
  start = perf_counter()
54
  images = lcm_text_to_image.generate(lcm_diffusion_setting)
55
  latency = perf_counter() - start