From ff8432500b991b73600f82847da0afb7f7b1807b Mon Sep 17 00:00:00 2001
From: komaksym <kovalmaksym2@gmail.com>
Date: Thu, 6 Mar 2025 05:27:21 +0100
Subject: [PATCH] fixed finetune requirements.txt

---
 finetune/install.sh       | 4 ++++
 finetune/requirements.txt | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)
 create mode 100755 finetune/install.sh

diff --git a/finetune/install.sh b/finetune/install.sh
new file mode 100755
index 0000000..250a1e5
--- /dev/null
+++ b/finetune/install.sh
@@ -0,0 +1,4 @@
+#!/bin/bash
+pip install -r requirements.txt
+DS_BUILD_CPU_ADAM=1 pip install deepspeed==0.14.5
+
diff --git a/finetune/requirements.txt b/finetune/requirements.txt
index bc1d43e..d382942 100644
--- a/finetune/requirements.txt
+++ b/finetune/requirements.txt
@@ -2,9 +2,9 @@ torch==2.0.1
 tokenizers==0.14.0
 transformers==4.35.0
 accelerate==0.24.1
+numpy==1.26.4
 attrdict
 tqdm
 
-deepspeed==0.12.2
 datasets
 tensorboardX