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

diffusers better model autodetect

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-22 10:28:10 -04:00
parent 4518cfb3b1
commit 4e7ad0e15a
7 changed files with 70 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared_items.py
+15
View File
@@ -23,3 +23,18 @@ def list_crossattention():
"Sub-quadratic",
"Split attention"
]
def get_pipelines():
import diffusers
return {
'Autodetect': None,
'Stable Diffusion': diffusers.StableDiffusionPipeline,
'Stable Diffusion Img2Img': diffusers.StableDiffusionImg2ImgPipeline,
'Stable Diffusion Instruct': diffusers.StableDiffusionInstructPix2PixPipeline,
'Stable Diffusion Upscale': diffusers.StableDiffusionUpscalePipeline,
'Stable Diffusion XL': diffusers.StableDiffusionXLPipeline,
'Stable Diffusion XL Img2Img': diffusers.StableDiffusionXLImg2ImgPipeline,
'Stable Diffusion XL Inpaint': diffusers.StableDiffusionXLInpaintPipeline,
'Stable Diffusion XL Instruct': diffusers.StableDiffusionXLInstructPix2PixPipeline,
# 'Kandinsky V1', 'Kandinsky V2', 'DeepFloyd IF', 'Shap-E', 'Kandinsky V1 Img2Img', 'Kandinsky V2 Img2Img', 'DeepFloyd IF Img2Img', 'Shap-E Img2Img',
}