rupeshs commited on
Commit
ea39d32
1 Parent(s): d90abc2

added safety filter tiny vae

Browse files
Files changed (1) hide show
  1. frontend/webui/hf_demo.py +2 -0
frontend/webui/hf_demo.py CHANGED
@@ -50,6 +50,8 @@ def predict(
50
  lcm_diffusion_setting.inference_steps = steps
51
  lcm_diffusion_setting.seed = seed
52
  lcm_diffusion_setting.use_seed = use_seed
 
 
53
  lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
54
  lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
55
  lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
 
50
  lcm_diffusion_setting.inference_steps = steps
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 = True
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.use_openvino = True if is_openvino_device() else False