From 79d6e1d6a0e65e8f1e4a58521e97f2766dc7216e Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 14 Sep 2023 15:57:31 +0300 Subject: [PATCH] IPEX fix diffusers 0.21 lazy load --- modeling/ipex/diffusers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modeling/ipex/diffusers.py b/modeling/ipex/diffusers.py index 3435abe1..4c39896e 100644 --- a/modeling/ipex/diffusers.py +++ b/modeling/ipex/diffusers.py @@ -1,6 +1,7 @@ import torch import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import -import diffusers #0.20.2 # pylint: disable=import-error +import diffusers #0.21.1 # pylint: disable=import-error +from diffusers.models.attention_processor import Attention # pylint: disable=protected-access, missing-function-docstring, line-too-long @@ -17,7 +18,7 @@ class SlicedAttnProcessor: # pylint: disable=too-few-public-methods def __init__(self, slice_size): self.slice_size = slice_size - def __call__(self, attn: diffusers.models.attention_processor.Attention, hidden_states, encoder_hidden_states=None, attention_mask=None): # pylint: disable=too-many-statements, too-many-locals, too-many-branches + def __call__(self, attn: Attention, hidden_states, encoder_hidden_states=None, attention_mask=None): # pylint: disable=too-many-statements, too-many-locals, too-many-branches residual = hidden_states input_ndim = hidden_states.ndim