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

refactor onnx and olive

Browse Source
This commit is contained in:
Seunghoon Lee
2023-11-04 23:46:20 +09:00
parent b70258c926
commit 6507491d8f
27 changed files with 1185 additions and 342 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared_items.py
+3 -4
View File
@@ -26,8 +26,7 @@ def list_crossattention():
def get_pipelines():
import diffusers
from modules.onnx import OnnxStableDiffusionPipeline
from modules.olive import OlivePipeline, is_available as is_olive_available
from modules.onnx import OnnxAutoPipeline
from installer import log
pipelines = { # note: not all pipelines can be used manually as they require prior pipeline next to decoder pipeline
'Autodetect': None,
@@ -40,8 +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': OnnxStableDiffusionPipeline,
'ONNX Stable Diffusion with Olive': OlivePipeline,
'ONNX Stable Diffusion': OnnxAutoPipeline,
'ONNX Stable Diffusion with Olive': OnnxAutoPipeline,
'Latent Consistency Model': getattr(diffusers, 'LatentConsistencyModelPipeline', None),
'PixArt Alpha': getattr(diffusers, 'PixArtAlphaPipeline', None),
'UniDiffuser': getattr(diffusers, 'UniDiffuserPipeline', None),