From be338ec53aa966afaabd3af95975358c6cce7611 Mon Sep 17 00:00:00 2001
From: Anubha Jain <41411445+Anubha98@users.noreply.github.com>
Date: Tue, 20 Aug 2024 14:05:50 +0530
Subject: [PATCH] NLP env vulnerabilities fix (#3271)

Co-authored-by: Anubha Jain <anubhajain@microsoft.com>
---
 .../finetune_acft_hf_nlp/environments/acpt/context/Dockerfile  | 2 +-
 .../environments/acpt/context/requirements.txt                 | 3 ++-
 2 files changed, 3 insertions(+), 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
index 6bf3c56bd7..2bb953f6eb 100644
--- 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
index 215091bacc..557aa4f398 100644
--- 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