Sujit Pal commited on
Commit
e2b6044
1 Parent(s): a78bf29

fix: added torch and torchvision

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -4,3 +4,5 @@ transformers
4
  jax
5
  jaxlib
6
  flax
 
 
 
4
  jax
5
  jaxlib
6
  flax
7
+ torch==1.9.0
8
+ torchvision==0.10.0