Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
transformers[torch]
|
2 |
-
torch
|
3 |
-
torchvision
|
4 |
-
torchaudio
|
5 |
einops
|
6 |
peft
|
7 |
omegaconf
|
|
|
1 |
transformers[torch]
|
2 |
+
torch==2.1.2
|
3 |
+
torchvision==2.1.2
|
4 |
+
torchaudio==2.1.2
|
5 |
einops
|
6 |
peft
|
7 |
omegaconf
|