From 7b208253a974f38c3605b03eb2a6e8cf6afc9d08 Mon Sep 17 00:00:00 2001 From: Jug Date: Mon, 6 Mar 2023 21:30:55 +0000 Subject: [PATCH] Fix for inpainting when loading "ALL MODELS" --- worker/bridge_data/stable_diffusion.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/worker/bridge_data/stable_diffusion.py b/worker/bridge_data/stable_diffusion.py index 745a87b8..48c9a572 100644 --- a/worker/bridge_data/stable_diffusion.py +++ b/worker/bridge_data/stable_diffusion.py @@ -118,6 +118,10 @@ def get_all_models(self): models = sorted(list(set(models))) + # Add inpainting model if inpainting is enabled + if self.allow_painting: + models.append("stable_diffusion_inpainting") + # Save our models to our environmental variable os.environ["ALL_MODELS"] = ",".join(models)