Spaces:
Running
on
A100
Running
on
A100
xformers not on osx
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
diffusers==0.23.0
|
2 |
transformers==4.34.1
|
3 |
gradio==3.50.2
|
4 |
-
--extra-index-url https://download.pytorch.org/whl/cu121
|
5 |
torch==2.1.0
|
6 |
fastapi==0.104.0
|
7 |
uvicorn==0.23.2
|
@@ -10,4 +10,4 @@ accelerate==0.24.0
|
|
10 |
compel==2.0.2
|
11 |
controlnet-aux==0.0.7
|
12 |
peft==0.6.0
|
13 |
-
xformers
|
|
|
1 |
diffusers==0.23.0
|
2 |
transformers==4.34.1
|
3 |
gradio==3.50.2
|
4 |
+
--extra-index-url https://download.pytorch.org/whl/cu121;
|
5 |
torch==2.1.0
|
6 |
fastapi==0.104.0
|
7 |
uvicorn==0.23.2
|
|
|
10 |
compel==2.0.2
|
11 |
controlnet-aux==0.0.7
|
12 |
peft==0.6.0
|
13 |
+
xformers; sys_platform != 'darwin' or platform_machine != 'arm64'
|