fffiloni commited on
Commit
3e92eb2
1 Parent(s): 4df05c6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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