Skip to content

Commit f7d379d

Browse files
authored
Fix compatibility with diffusers < 0.25.0 (#952)
1 parent 93ca007 commit f7d379d

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

optimum/intel/openvino/modeling_diffusion.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
import openvino
2929
import torch
3030
from diffusers.configuration_utils import ConfigMixin
31-
from diffusers.models.autoencoders.vae import DiagonalGaussianDistribution
3231
from diffusers.pipelines import (
3332
AutoPipelineForImage2Image,
3433
AutoPipelineForInpainting,
@@ -63,6 +62,7 @@
6362
)
6463

6564
from ...exporters.openvino import main_export
65+
from ..utils.import_utils import is_diffusers_version
6666
from .configuration import OVConfig, OVQuantizationMethod, OVWeightQuantizationConfig
6767
from .loaders import OVTextualInversionLoaderMixin
6868
from .modeling_base import OVBaseModel
@@ -76,6 +76,12 @@
7676
)
7777

7878

79+
if is_diffusers_version(">=", "0.25.0"):
80+
from diffusers.models.autoencoders.vae import DiagonalGaussianDistribution
81+
else:
82+
from diffusers.models.vae import DiagonalGaussianDistribution
83+
84+
7985
core = Core()
8086

8187
logger = logging.getLogger(__name__)

0 commit comments

Comments
 (0)