adjusted dep version
@@ -32,10 +32,11 @@ requirements:
- umap-learn >=0.5.0
- cython
- sentence-transformers
- - transformers
- - pytorch >=1.4.0,<1.7.1
- - sentencepiece
+ - pytorch >=1.6.0,<1.7.1
- torchvision
+ - sentencepiece
+ - tokenizers >=0.10.3
+ - transformers >=4.6.0,<5.0.0
test:
imports:
