VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup IPEX libs

Browse Source
This commit is contained in:
Disty0
2023-12-05 19:52:31 +03:00
parent 3d70137d31
commit a9c6182b3f
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
library/ipex/__init__.py
View File

@@ -4,7 +4,6 @@ import contextlib
import torch import torch
import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
from .hijacks import ipex_hijacks from .hijacks import ipex_hijacks
from .attention import attention_init
# pylint: disable=protected-access, missing-function-docstring, line-too-long # pylint: disable=protected-access, missing-function-docstring, line-too-long
@@ -166,7 +165,11 @@ def ipex_init(): # pylint: disable=too-many-statements
ipex_hijacks() ipex_hijacks()
if not torch.xpu.has_fp64_dtype(): if not torch.xpu.has_fp64_dtype():
try:
from .attention import attention_init
attention_init() attention_init()
except Exception: # pylint: disable=broad-exception-caught
pass
try: try:
from .diffusers import ipex_diffusers from .diffusers import ipex_diffusers
ipex_diffusers() ipex_diffusers()