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

@@ -3,6 +3,13 @@ import os
import regex
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 open_clip
from library import sdxl_model_util, sdxl_train_util, train_util