diff --git a/cv/aml-cli-v2/data-science/environment/Dockerfile b/cv/aml-cli-v2/data-science/environment/Dockerfile index 31f3d98..540bab9 100644 --- a/cv/aml-cli-v2/data-science/environment/Dockerfile +++ b/cv/aml-cli-v2/data-science/environment/Dockerfile @@ -26,7 +26,7 @@ RUN pip install -r requirements.txt # RUN python -m pip install azureml-defaults==1.41.0 \ # mlflow==1.25.1 \ # azureml-mlflow==1.41.0 \ -# transformers==4.18.0 \ +# transformers==4.30.0 \ # psutil==5.9.0 # add ndv4-topo.xml diff --git a/cv/aml-cli-v2/data-science/environment/requirements.txt b/cv/aml-cli-v2/data-science/environment/requirements.txt index f71e316..c648d6b 100644 --- a/cv/aml-cli-v2/data-science/environment/requirements.txt +++ b/cv/aml-cli-v2/data-science/environment/requirements.txt @@ -1,7 +1,7 @@ # for local testing (cpu) torchvision==0.12.0 torch==1.13.1 -transformers==4.18.0 +transformers==4.30.0 # for metrics reporting/plotting mlflow==2.3.1 diff --git a/cv/aml-cli-v2/data-science/requirements-tests.txt b/cv/aml-cli-v2/data-science/requirements-tests.txt index e2af0c2..e419cc3 100644 --- a/cv/aml-cli-v2/data-science/requirements-tests.txt +++ b/cv/aml-cli-v2/data-science/requirements-tests.txt @@ -3,7 +3,7 @@ # CV packages torchvision==0.12.0 torch==1.13.1 -transformers==4.18.0 +transformers==4.30.0 # for metrics reporting/plotting mlflow==2.3.1 diff --git a/nlp/aml-cli-v2/data-science/environments/training/requirements.txt b/nlp/aml-cli-v2/data-science/environments/training/requirements.txt index 1f505dd..d296fb3 100644 --- a/nlp/aml-cli-v2/data-science/environments/training/requirements.txt +++ b/nlp/aml-cli-v2/data-science/environments/training/requirements.txt @@ -2,7 +2,7 @@ # torchvision==0.12.0 # torch==1.11.0 pytorch_lightning==1.6.4 -transformers==4.18.0 +transformers==4.30.0 datasets==2.3.2 rouge_score==0.0.4 sentencepiece==0.1.96 diff --git a/nlp/python-sdk-v2/data-science/environments/training/requirements.txt b/nlp/python-sdk-v2/data-science/environments/training/requirements.txt index 173fe3f..a7dc348 100644 --- a/nlp/python-sdk-v2/data-science/environments/training/requirements.txt +++ b/nlp/python-sdk-v2/data-science/environments/training/requirements.txt @@ -2,7 +2,7 @@ # torchvision==0.12.0 # torch==1.11.0 pytorch_lightning==1.6.4 -transformers==4.18.0 +transformers==4.30.0 datasets==2.3.2 rouge_score==0.0.4 sentencepiece==0.1.96