rupeshs commited on
Commit
f4e968d
·
1 Parent(s): 6144792

size update

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +3 -3
frontend/webui/hf_demo.py CHANGED
@@ -53,8 +53,8 @@ def predict(
53
  lcm_diffusion_setting.use_tiny_auto_encoder = True
54
  # lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
55
  # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
56
- lcm_diffusion_setting.image_width = 320
57
- lcm_diffusion_setting.image_height = 320
58
  lcm_diffusion_setting.use_openvino = True
59
  lcm_diffusion_setting.use_tiny_auto_encoder = True
60
  pprint(lcm_diffusion_setting.model_dump())
@@ -63,7 +63,7 @@ def predict(
63
  images = lcm_text_to_image.generate(lcm_diffusion_setting)
64
  latency = perf_counter() - start
65
  print(f"Latency: {latency:.2f} seconds")
66
- return images[0].resize([512, 512], PIL.Image.ANTIALIAS)
67
 
68
 
69
  css = """
 
53
  lcm_diffusion_setting.use_tiny_auto_encoder = True
54
  # lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
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())
 
63
  images = lcm_text_to_image.generate(lcm_diffusion_setting)
64
  latency = perf_counter() - start
65
  print(f"Latency: {latency:.2f} seconds")
66
+ return images[0] # .resize([512, 512], PIL.Image.ANTIALIAS)
67
 
68
 
69
  css = """