Skip to content

Commit 51172d0

Browse files
authored
[Update]: Updated environment of acft_multimodal (#4225)
* [Update]: Updated environment of acft_multimodal * [Update]: Base image update with torch upgrade
1 parent 4c43160 commit 51172d0

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

assets/training/finetune_acft_multimodal/environments/acpt_multimodal/context/Dockerfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# PTCA image
2-
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch222:{{latest-image-tag:biweekly\.\d{6}\.\d{1}.*}}
2+
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch260:{{latest-image-tag:biweekly\.\d{6}\.\d{1}.*}}
33

44
USER root
55
RUN apt-get -y update && apt-get install -y expat
@@ -18,4 +18,6 @@ RUN pip install cryptography==44.0.1
1818
RUN pip install gunicorn==23.0.0
1919
RUN pip install onnxruntime==1.17.3
2020
RUN pip install onnx==1.17.0
21-
RUN pip install scikit-learn==1.5.0
21+
RUN pip install scikit-learn==1.5.0
22+
RUN pip install azureml-acft-common-components=={{latest-pypi-version}}
23+
RUN pip install azureml-acft-accelerator=={{latest-pypi-version}}

assets/training/finetune_acft_multimodal/environments/acpt_multimodal/context/requirements.txt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
1-
azureml-acft-accelerator=={{latest-pypi-version}}
21
# azureml-evaluate-mlflow=={{latest-pypi-version}}
3-
azureml-acft-common-components=={{latest-pypi-version}}
42
azureml-acft-contrib-hf-nlp=={{latest-pypi-version}}
53
# Temporarily commenting azureml_acft_multimodal_components since it is not supported in python 3.10 and we need to
64
# upgrade to python 3.10 to address scikit-learn vulnerability(version 1.5.0) which requires python>=3.9
@@ -10,7 +8,7 @@ azureml-acft-contrib-hf-nlp=={{latest-pypi-version}}
108
mltable==1.5.0
119
transformers==4.48.0
1210
peft==0.4.0
13-
deepspeed==0.15.2
11+
deepspeed==0.16.9
1412
optimum==1.23.3
1513
accelerate==0.25.0
1614
diffusers==0.24.0

0 commit comments

Comments
 (0)