From 53019de30f371cf7d95ddf37af823fc86e5f11fe Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sat, 18 Nov 2023 09:22:10 -0500
Subject: [PATCH] todo cleanup

---
 modules/sd_models.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 6a5a40fef..ab1188c60 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -989,6 +989,26 @@ def set_diffuser_pipe(pipe, new_pipe_type):
     has_accelerate = getattr(pipe, "has_accelerate", None)
     embedding_db = getattr(pipe, "embedding_db", None)
 
+    # TODO implement alternative diffusion pipelines
+    """
+    from collections import OrderedDict
+    AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
+        [
+            ("stable-diffusion", diffusers.StableDiffusionPipeline),
+            ("stable-diffusion-xl", diffusers.StableDiffusionXLPipeline),
+            ("if", diffusers.IFPipeline),
+            ("kandinsky", diffusers.KandinskyCombinedPipeline),
+            ("kandinsky22", diffusers.KandinskyV22CombinedPipeline),
+            ("stable-diffusion-controlnet", diffusers.StableDiffusionControlNetPipeline),
+            ("stable-diffusion-xl-controlnet", diffusers.StableDiffusionXLControlNetPipeline),
+            ("wuerstchen", diffusers.WuerstchenCombinedPipeline),
+            ("lcm", diffusers.LatentConsistencyModelPipeline),
+            ("pixart", diffusers.PixArtAlphaPipeline),
+        ]
+    )
+    diffusers.pipelines.auto_pipeline.AUTO_TEXT2IMAGE_PIPELINES_MAPPING = AUTO_TEXT2IMAGE_PIPELINES_MAPPING
+    """
+
     if shared.opts.diffusers_force_inpaint:
         if new_pipe_type == DiffusersTaskType.IMAGE_2_IMAGE:
             new_pipe_type = DiffusersTaskType.INPAINTING # sdxl may work better with init mask