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

support more backends

Browse Source
This commit is contained in:
Seunghoon Lee
2023-10-25 19:18:14 +09:00
parent c327331312
commit df021b3982
6 changed files with 39 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared_items.py
+3 -2
View File
@@ -26,7 +26,7 @@ def list_crossattention():
def get_pipelines():
import diffusers
from modules.olive import OlivePipeline
from modules.olive import OnnxStableDiffusionPipeline, OlivePipeline
from installer import log
pipelines = { # note: not all pipelines can be used manually as they require prior pipeline next to decoder pipeline
'Autodetect': None,
@@ -39,7 +39,8 @@ def get_pipelines():
'Stable Diffusion XL Img2Img': getattr(diffusers, 'StableDiffusionXLImg2ImgPipeline', None),
'Stable Diffusion XL Inpaint': getattr(diffusers, 'StableDiffusionXLInpaintPipeline', None),
'Stable Diffusion XL Instruct': getattr(diffusers, 'StableDiffusionXLInstructPix2PixPipeline', None),
'ONNX Stable Diffusion (Olive)': OlivePipeline,
'ONNX Stable Diffusion': OnnxStableDiffusionPipeline,
'ONNX Stable Diffusion with Olive': OlivePipeline,
'Latent Consistency Model': getattr(diffusers, 'LatentConsistencyModelPipeline', None),
'PixArt Alpha': getattr(diffusers, 'PixArtAlphaPipeline', None),
'UniDiffuser': getattr(diffusers, 'UniDiffuserPipeline', None),