rupeshs commited on
Commit
a704e4e
·
1 Parent(s): 4b51ee4

model update

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +4 -3
frontend/webui/hf_demo.py CHANGED
@@ -41,8 +41,9 @@ def predict(
41
  ):
42
  print(f"prompt - {prompt}")
43
  lcm_diffusion_setting = LCMDiffusionSetting()
 
44
  lcm_diffusion_setting.diffusion_task = DiffusionTask.text_to_image.value
45
- lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/hyper-sd-sdxl-1-step"
46
  lcm_diffusion_setting.use_lcm_lora = False
47
  lcm_diffusion_setting.prompt = prompt
48
  lcm_diffusion_setting.guidance_scale = 1.0
@@ -50,7 +51,7 @@ def predict(
50
  lcm_diffusion_setting.seed = seed
51
  lcm_diffusion_setting.use_seed = use_seed
52
  lcm_diffusion_setting.use_safety_checker = True
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
@@ -63,7 +64,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]
67
 
68
 
69
  css = """
 
41
  ):
42
  print(f"prompt - {prompt}")
43
  lcm_diffusion_setting = LCMDiffusionSetting()
44
+ lcm_diffusion_setting.lcm_model_id = "rupeshs/hyper-sd-sdxl-1-step"
45
  lcm_diffusion_setting.diffusion_task = DiffusionTask.text_to_image.value
46
+ lcm_diffusion_setting.openvino_lcm_model_id = "rupeshs/sd-turbo-openvino"
47
  lcm_diffusion_setting.use_lcm_lora = False
48
  lcm_diffusion_setting.prompt = prompt
49
  lcm_diffusion_setting.guidance_scale = 1.0
 
51
  lcm_diffusion_setting.seed = seed
52
  lcm_diffusion_setting.use_seed = use_seed
53
  lcm_diffusion_setting.use_safety_checker = True
54
+ lcm_diffusion_setting.use_tiny_auto_encoder = False
55
  # lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
56
  # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
57
  lcm_diffusion_setting.image_width = 512
 
64
  images = lcm_text_to_image.generate(lcm_diffusion_setting)
65
  latency = perf_counter() - start
66
  print(f"Latency: {latency:.2f} seconds")
67
+ return images[0] # .resize([512, 512], PIL.Image.ANTIALIAS)
68
 
69
 
70
  css = """