changed deps
@@ -26,12 +26,16 @@ requirements:
- pandas >=1.1.5
- plotly >=4.7.0
- python >=3.7
- - sentence-transformers >=0.4.1,
- pyyaml <6.0
- scikit-learn >=0.22.2.post1
- tqdm >=4.41.1
- umap-learn >=0.5.0
- cython
+ - sentence-transformers >=0.4.1,
+ - transformers =3.5.1
+ - pytorch >=1.4.0,<1.7.1
+ - sentencepiece
+ - torchvision
test:
imports:
