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

@@ -4,6 +4,13 @@
import math
import os
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 diffusers
from transformers import CLIPTextModel, CLIPTokenizer, CLIPTextConfig, logging
from diffusers import AutoencoderKL, DDIMScheduler, StableDiffusionPipeline # , UNet2DConditionModel