Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
torch
|
| 2 |
transformers>=4.0.0
|
| 3 |
streamlit==1.41.1
|
| 4 |
datasets
|
|
@@ -13,3 +13,4 @@ spaces
|
|
| 13 |
diffusers
|
| 14 |
sentencepiece
|
| 15 |
accelerate
|
|
|
|
|
|
| 1 |
+
torch>=2.0.1
|
| 2 |
transformers>=4.0.0
|
| 3 |
streamlit==1.41.1
|
| 4 |
datasets
|
|
|
|
| 13 |
diffusers
|
| 14 |
sentencepiece
|
| 15 |
accelerate
|
| 16 |
+
torchvision
|