apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

wrap some diffusers imports for version safety

This commit is contained in:
Sean Sube 2023-12-28 22:49:17 -06:00
parent f28fdda47a
commit e7f7bbc9ea
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/diffusers/patches/vae.py
View File

@ -4,9 +4,8 @@ from typing import Union
import numpy as np
import torch
from diffusers import OnnxRuntimeModel
from diffusers.models.autoencoder_kl import AutoencoderKLOutput
from diffusers.models.vae import DecoderOutput
from diffusers.pipelines.onnx_utils import ORT_TO_NP_TYPE
from ..version_safe_diffusers import AutoencoderKLOutput, DecoderOutput
from ...server import ServerContext

12
api/onnx_web/diffusers/version_safe_diffusers.py
View File

@ -5,7 +5,9 @@ from packaging import version
is_diffusers_0_15 = version.parse(
version.parse(diffusers.__version__).base_version
) >= version.parse("0.15")
is_diffusers_0_24 = version.parse(
version.parse(diffusers.__version__).base_version
) >= version.parse("0.24")
try:
from diffusers import DEISMultistepScheduler
@ -32,3 +34,11 @@ if is_diffusers_0_15:
from diffusers.models.attention_processor import AttnProcessor
else:
from diffusers.models.cross_attention import CrossAttnProcessor as AttnProcessor
if is_diffusers_0_24:
from diffusers.models.modeling_outputs import AutoencoderKLOutput
from diffusers.models.autoencoders.vae import DecoderOutput
else:
from diffusers.models.autoencoder_kl import AutoencoderKLOutput
from diffusers.models.vae import DecoderOutput