Sofian Hadiwijaya
commited on
Commit
•
955f529
1
Parent(s):
fe4282c
change cu to cu 117
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
-
torch==2.0.1 --index-url https://download.pytorch.org/whl/
|
| 2 |
-
torchvision==0.15.2 --index-url https://download.pytorch.org/whl/
|
| 3 |
-
torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/
|
| 4 |
diffusers==0.27.2
|
| 5 |
accelerate==0.28.0
|
| 6 |
tensorflow==2.12.0
|
|
|
|
| 1 |
+
torch==2.0.1 --index-url https://download.pytorch.org/whl/cu117
|
| 2 |
+
torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
|
| 3 |
+
torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117
|
| 4 |
diffusers==0.27.2
|
| 5 |
accelerate==0.28.0
|
| 6 |
tensorflow==2.12.0
|