From b7ba565956fe61153064f288862daf1896de5393 Mon Sep 17 00:00:00 2001
From: DejianYang <dejianyang@deepseek.com>
Date: Tue, 16 Apr 2024 10:16:07 +0800
Subject: [PATCH] Update requirements.txt for finetune

---
 finetune/requirements.txt | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/finetune/requirements.txt b/finetune/requirements.txt
index 07a0685..bc1d43e 100644
--- a/finetune/requirements.txt
+++ b/finetune/requirements.txt
@@ -1,10 +1,10 @@
-torch>=2.0.1
-tokenizers>=0.14.0
+torch==2.0.1
+tokenizers==0.14.0
 transformers==4.35.0
-accelerate
+accelerate==0.24.1
 attrdict
 tqdm
 
-deepspeed
+deepspeed==0.12.2
 datasets
 tensorboardX