Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

IPEX fix diffusers hijacks

Browse Source
This commit is contained in:
Disty0
2025-04-30 22:35:52 +03:00
parent 149b6695d9
commit 5cb8bad576
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/intel/ipex/diffusers.py
+3 -4
View File
@@ -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