rupeshs commited on
Commit
5a61b96
·
1 Parent(s): 992f222

updated to sdxs

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +4 -4
frontend/webui/hf_demo.py CHANGED
@@ -42,7 +42,7 @@ def predict(
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
@@ -53,10 +53,10 @@ 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 = 768
57
- lcm_diffusion_setting.image_height = 768
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)
62
  start = perf_counter()
 
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/sdxs-512-0.9-orig-vae"
46
  lcm_diffusion_setting.use_lcm_lora = False
47
  lcm_diffusion_setting.prompt = prompt
48
  lcm_diffusion_setting.guidance_scale = 1.0
 
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 = False
59
+ lcm_diffusion_setting.use_tiny_auto_encoder = False
60
  pprint(lcm_diffusion_setting.model_dump())
61
  lcm_text_to_image.init(lcm_diffusion_setting=lcm_diffusion_setting)
62
  start = perf_counter()