Skip to content

Commit 0be681e

Browse files
committed
feat(intel): add diffusers support
1 parent d825821 commit 0be681e

File tree

3 files changed

+73
-0
lines changed

3 files changed

+73
-0
lines changed

backend/python/diffusers/Makefile

+4
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@ ifeq ($(BUILD_TYPE), hipblas)
44
export CONDA_ENV_PATH = "diffusers-rocm.yml"
55
endif
66

7+
ifneq (,$(findstring sycl,$(BUILD_TYPE)))
8+
export CONDA_ENV_PATH = "diffusers-intel.yml"
9+
endif
10+
711
.PHONY: diffusers
812
diffusers:
913
@echo "Installing $(CONDA_ENV_PATH)..."
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,64 @@
1+
name: diffusers
2+
channels:
3+
- defaults
4+
dependencies:
5+
- _libgcc_mutex=0.1=main
6+
- _openmp_mutex=5.1=1_gnu
7+
- bzip2=1.0.8=h7b6447c_0
8+
- ca-certificates=2023.08.22=h06a4308_0
9+
- ld_impl_linux-64=2.38=h1181459_1
10+
- libffi=3.4.4=h6a678d5_0
11+
- libgcc-ng=11.2.0=h1234567_1
12+
- libgomp=11.2.0=h1234567_1
13+
- libstdcxx-ng=11.2.0=h1234567_1
14+
- libuuid=1.41.5=h5eee18b_0
15+
- ncurses=6.4=h6a678d5_0
16+
- openssl=3.0.11=h7f8727e_2
17+
- pip=23.2.1=py311h06a4308_0
18+
- python=3.11.5=h955ad1f_0
19+
- readline=8.2=h5eee18b_0
20+
- setuptools=68.0.0=py311h06a4308_0
21+
- sqlite=3.41.2=h5eee18b_0
22+
- tk=8.6.12=h1ccaba5_0
23+
- tzdata=2023c=h04d1e81_0
24+
- wheel=0.41.2=py311h06a4308_0
25+
- xz=5.4.2=h5eee18b_0
26+
- zlib=1.2.13=h5eee18b_0
27+
- pip:
28+
- --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
29+
- accelerate>=0.11.0
30+
- certifi==2023.7.22
31+
- charset-normalizer==3.3.0
32+
- compel==2.0.2
33+
- diffusers==0.24.0
34+
- filelock==3.12.4
35+
- fsspec==2023.9.2
36+
- grpcio==1.59.0
37+
- huggingface-hub>=0.19.4
38+
- idna==3.4
39+
- importlib-metadata==6.8.0
40+
- jinja2==3.1.2
41+
- markupsafe==2.1.3
42+
- mpmath==1.3.0
43+
- networkx==3.1
44+
- numpy==1.26.0
45+
- omegaconf
46+
- packaging==23.2
47+
- pillow==10.0.1
48+
- protobuf==4.24.4
49+
- psutil==5.9.5
50+
- pyparsing==3.1.1
51+
- pyyaml==6.0.1
52+
- regex==2023.10.3
53+
- requests==2.31.0
54+
- safetensors==0.4.0
55+
- sympy==1.12
56+
- torch==2.1.0a0
57+
- intel-extension-for-pytorch==2.1.10+xpu
58+
- tqdm==4.66.1
59+
- transformers>=4.25.1
60+
- triton==2.1.0
61+
- typing-extensions==4.8.0
62+
- urllib3==2.0.6
63+
- zipp==3.17.0
64+
prefix: /opt/conda/envs/diffusers

backend/python/diffusers/run.sh

+5
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,11 @@ export PATH=$PATH:/opt/conda/bin
88
# Activate conda environment
99
source activate diffusers
1010

11+
if [ -d "/opt/intel" ]; then
12+
source /opt/intel/oneapi/compiler/latest/env/vars.sh
13+
source /opt/intel/oneapi/mkl/latest/env/vars.sh
14+
fi
15+
1116
# get the directory where the bash script is located
1217
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
1318

0 commit comments

Comments
 (0)