added safety filter tiny vae
Browse files
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
|