File tree 1 file changed +7
-1
lines changed
1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 28
28
import openvino
29
29
import torch
30
30
from diffusers .configuration_utils import ConfigMixin
31
- from diffusers .models .autoencoders .vae import DiagonalGaussianDistribution
32
31
from diffusers .pipelines import (
33
32
AutoPipelineForImage2Image ,
34
33
AutoPipelineForInpainting ,
63
62
)
64
63
65
64
from ...exporters .openvino import main_export
65
+ from ..utils .import_utils import is_diffusers_version
66
66
from .configuration import OVConfig , OVQuantizationMethod , OVWeightQuantizationConfig
67
67
from .loaders import OVTextualInversionLoaderMixin
68
68
from .modeling_base import OVBaseModel
76
76
)
77
77
78
78
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
+
79
85
core = Core ()
80
86
81
87
logger = logging .getLogger (__name__ )
You can’t perform that action at this time.
0 commit comments