diff --git a/environment.yml b/environment.yml index 67535adc..e0691da8 100644 --- a/environment.yml +++ b/environment.yml @@ -3,13 +3,13 @@ channels: - conda-forge dependencies: - python=3.* - - numpy=1.21.* - - pytorch - - transformers - - sentence-transformers - - fastapi - - uvicorn - - pyyaml - - pytest - - pillow - - torchvision \ No newline at end of file + - numpy=1.* + - pytorch=1.* + - transformers=4.* + - sentence-transformers=2.0.0 + - fastapi=0.* + - uvicorn=0.* + - pyyaml=5.* + - pytest=6.* + - pillow=8.* + - torchvision=0.* \ No newline at end of file