From cb821701875f80272b39e50d47fcf3d48cb2d175 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Mon, 17 Oct 2022 21:22:05 +0530 Subject: [PATCH] Fix the bug where custom models weren't getting picked up --- ui/sd_internal/task_manager.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ui/sd_internal/task_manager.py b/ui/sd_internal/task_manager.py index 00364e0c..cdefc53a 100644 --- a/ui/sd_internal/task_manager.py +++ b/ui/sd_internal/task_manager.py @@ -282,6 +282,7 @@ def render(req : ImageRequest): r.save_to_disk_path = req.save_to_disk_path r.use_upscale: str = req.use_upscale r.use_face_correction = req.use_face_correction + r.use_stable_diffusion_model = req.use_stable_diffusion_model r.show_only_filtered_image = req.show_only_filtered_image r.output_format = req.output_format