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

@@ -10,6 +10,13 @@ import toml
from tqdm import tqdm
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
from accelerate.utils import set_seed
from diffusers import DDPMScheduler