adamantix commited on
Commit
548e443
·
verified ·
1 Parent(s): 4cfdc1d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch
 
2
  transformers
3
  xgboost
4
  git+https://github.com/mlfoundations/open_clip.git
 
1
+ torch==2.1.2+cpu --index-url https://download.pytorch.org/whl/cpu
2
+ torchvision==0.16.2+cpu --index-url https://download.pytorch.org/whl/cpu
3
  transformers
4
  xgboost
5
  git+https://github.com/mlfoundations/open_clip.git