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

@@ -1,5 +1,12 @@
import argparse
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 library import sdxl_model_util, sdxl_train_util, train_util
import train_network