Deduplicate ipex initialization code

This commit is contained in:
Aarni Koskela
2024-01-18 18:23:21 +02:00
parent d2a99a19d4
commit 6f3f701d3d
17 changed files with 70 additions and 113 deletions

View File

@@ -5,15 +5,9 @@ import math
import os
import torch
try:
import intel_extension_for_pytorch as ipex
from library.ipex_interop import init_ipex
if torch.xpu.is_available():
from library.ipex import ipex_init
ipex_init()
except Exception:
pass
init_ipex()
import diffusers
from transformers import CLIPTextModel, CLIPTokenizer, CLIPTextConfig, logging
from diffusers import AutoencoderKL, DDIMScheduler, StableDiffusionPipeline # , UNet2DConditionModel