Intel ARC support with IPEX

This commit is contained in:
Disty0
2023-09-19 18:05:05 +03:00
parent db7a28ac25
commit b64389c8a9
21 changed files with 928 additions and 0 deletions

View File

@@ -65,6 +65,13 @@ import re
import diffusers
import numpy as np
import torch
try:
import intel_extension_for_pytorch as ipex
if torch.xpu.is_available():
from library.ipex import ipex_init
ipex_init()
except Exception:
pass
import torchvision
from diffusers import (
AutoencoderKL,