From a8af6ccb73a615304bfc0115cfa0b5379b68a73d Mon Sep 17 00:00:00 2001 From: YeAnbang Date: Fri, 28 Jun 2024 03:58:29 +0000 Subject: [PATCH] fix torch colossalai version --- applications/ColossalChat/examples/requirements.txt | 2 +- applications/ColossalChat/requirements.txt | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/applications/ColossalChat/examples/requirements.txt b/applications/ColossalChat/examples/requirements.txt index 838590f4b..91f25a5cf 100644 --- a/applications/ColossalChat/examples/requirements.txt +++ b/applications/ColossalChat/examples/requirements.txt @@ -1,4 +1,4 @@ pandas>=1.4.1 sentencepiece -colossalai +colossalai==0.4.0 prompt_toolkit diff --git a/applications/ColossalChat/requirements.txt b/applications/ColossalChat/requirements.txt index 639ff6aef..acf4c64bf 100755 --- a/applications/ColossalChat/requirements.txt +++ b/applications/ColossalChat/requirements.txt @@ -2,8 +2,8 @@ transformers==4.36.2 tqdm datasets==2.14.7 loralib -colossalai>=0.3.7 -torch>=1.12.1 +colossalai==0.4.0 +torch>=2.1.0 langchain tokenizers fastapi