Aryeh Rotberg commited on
Commit
b365d37
·
1 Parent(s): 99b7e0e

Updated PyTorch to version 2.3.1+cu121 and TorchVision to 0.18.1+cu121 due to security reasons.

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -9,8 +9,8 @@ Pillow
9
  scikit_learn
10
  setuptools
11
  starlette
12
- torch==2.0.0+cu118
13
- torchvision==0.15.0+cu118
14
  tqdm
15
  python-multipart
16
  streamlit
 
9
  scikit_learn
10
  setuptools
11
  starlette
12
+ torch==2.3.1+cu121
13
+ torchvision==0.18.1+cu121
14
  tqdm
15
  python-multipart
16
  streamlit