diff --git a/requirements.txt b/requirements.txt index 65096865..ba44071d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ wheel # PyTorch # Temp. disabled the next line as it gets currently resolved to https://download.pytorch.org/whl/rocm3.8/torch-1.7.1%2Brocm3.8-cp38-cp38-linux_x86_64.whl #--find-links=https://download.pytorch.org/whl/torch_stable.html -torch>1.5,<1.9 +torch>1.5,<2.6 # progress bars in model download and training scripts tqdm # Accessing files from S3 directly. @@ -16,14 +16,14 @@ scipy>=1.3.2 sklearn # Metrics or logging related seqeval -mlflow<=1.13.1 +mlflow<=2.17.1 # huggingface repository -transformers==4.5.0 +transformers==4.38.0 # accessing dictionary elements with dot notation dotmap # for inference-rest-apis -Werkzeug==0.16.1 -flask<2.0.0 +Werkzeug==3.0.6 +flask<4.0.0 flask-restplus flask-cors dill # pickle extension for (de-)serialization