Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
8d9971c
1
Parent(s):
620c870
Add application file
Browse files- requirements.txt +4 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,6 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
|
| 3 |
-
torch==2.1.0
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
|
| 3 |
+
torch==2.1.0
|
| 4 |
+
torchaudio==2.1.0
|
| 5 |
+
torchvision==0.16.0
|
| 6 |
+
xformers==0.0.22.post7
|