NLP env vulnerabilities fix (#3271)

Co-authored-by: Anubha Jain <[email protected]>
Azure · Aug 20, 2024 · be338ec · be338ec
1 parent a0567e3
commit be338ec
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/assets/training/finetune_acft_hf_nlp/environments/acpt/context/Dockerfile b/assets/training/finetune_acft_hf_nlp/environments/acpt/context/Dockerfile
@@ -1,5 +1,5 @@
 #PTCA image
-FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py38-torch222:{{latest-image-tag:biweekly\.\d{6}\.\d{1}.*}}
+FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch222:{{latest-image-tag:biweekly\.\d{6}\.\d{1}.*}}
 
 USER root
 

diff --git a/assets/training/finetune_acft_hf_nlp/environments/acpt/context/requirements.txt b/assets/training/finetune_acft_hf_nlp/environments/acpt/context/requirements.txt
@@ -14,10 +14,11 @@ diffusers==0.26.3
 onnxruntime==1.17.3
 rouge-score==0.1.2
 sacrebleu==2.4.0
-bitsandbytes==0.42.0
+bitsandbytes==0.43.3
 einops==0.7.0
 aiohttp==3.9.4
 peft==0.8.2
 deepspeed==0.13.3
 trl==0.8.1
 tiktoken==0.6.0
+scipy==1.14.0