Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -7,5 +7,6 @@ ultralytics
|
|
| 7 |
salesforce-lavis
|
| 8 |
webdataset
|
| 9 |
pytorch_lightning
|
|
|
|
| 10 |
torchvision==0.16.2
|
| 11 |
gradio_imageslider
|
|
|
|
| 7 |
salesforce-lavis
|
| 8 |
webdataset
|
| 9 |
pytorch_lightning
|
| 10 |
+
torch==2.1.2
|
| 11 |
torchvision==0.16.2
|
| 12 |
gradio_imageslider
|