From 0324eb41036dc03fc2c5e88bd1b84bc19958c920 Mon Sep 17 00:00:00 2001 From: Yiqing-Zhou Date: Sat, 6 May 2023 21:05:53 +0800 Subject: [PATCH] [code] update requirements --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 08a6d2c..d6d2532 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ datasets==2.11.0 -deepspeed==0.9.1 -pytorch-lightning==2.0.2 +pytorch-lightning[strategies]==2.0.2 +tensorboardX==2.6 torch==2.0.0 transformers==4.28.1