Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-04 03:50:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

todo cleanup

Browse Source
This commit is contained in:
Vladimir Mandic
2023-11-18 09:22:10 -05:00
parent b1ecc7e0d4
commit 53019de30f
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+20
View File
@@ -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