From 5cb8bad57673bf69104acf1e69efee71a05a9770 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 30 Apr 2025 22:35:52 +0300 Subject: [PATCH] IPEX fix diffusers hijacks --- modules/intel/ipex/diffusers.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/modules/intel/ipex/diffusers.py b/modules/intel/ipex/diffusers.py index 8a10aa546..fa725f2ad 100644 --- a/modules/intel/ipex/diffusers.py +++ b/modules/intel/ipex/diffusers.py @@ -1,6 +1,7 @@ from functools import wraps import torch import diffusers # pylint: disable=import-error +from diffusers.utils import torch_utils # pylint: disable=import-error, unused-import # noqa: F401 # pylint: disable=protected-access, missing-function-docstring, line-too-long @@ -61,13 +62,11 @@ def hidream_rope(pos: torch.Tensor, dim: int, theta: int) -> torch.Tensor: def ipex_diffusers(device_supports_fp64=False, can_allocate_plus_4gb=False): - # get around lazy imports - from diffusers.utils import torch_utils # pylint: disable=import-error, unused-import diffusers.utils.torch_utils.fourier_filter = fourier_filter if not device_supports_fp64: # get around lazy imports - from diffusers.models import transformers as diffusers_transformers # pylint: disable=import-error, unused-import - from diffusers.models import controlnets as diffusers_controlnets # pylint: disable=import-error, unused-import + from diffusers.models import transformers as diffusers_transformers # pylint: disable=import-error, unused-import # noqa: F401 + from diffusers.models import controlnets as diffusers_controlnets # pylint: disable=import-error, unused-import # noqa: F401 diffusers.models.embeddings.FluxPosEmbed = FluxPosEmbed diffusers.models.transformers.transformer_flux.FluxPosEmbed = FluxPosEmbed diffusers.models.controlnets.controlnet_flux.FluxPosEmbed = FluxPosEmbed