Browse Source

[fix]downgrade colossalai (#4840)

cloud/coati
flybird11111 1 year ago committed by GitHub
parent
commit
ed888183ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      applications/Chat/examples/requirements.txt
  2. 2
      applications/Chat/requirements-test.txt
  3. 2
      applications/Chat/requirements.txt

2
applications/Chat/examples/requirements.txt

@ -1,3 +1,3 @@
pandas>=1.4.1
sentencepiece
colossalai==0.3.1
git+https://github.com/hpcaitech/ColossalAI.git@ef4b99ebcda823c32e221383ac9365a82697cd5d#egg=colossalai

2
applications/Chat/requirements-test.txt

@ -1,2 +1,2 @@
pytest
colossalai==0.3.1
git+https://github.com/hpcaitech/ColossalAI.git@ef4b99ebcda823c32e221383ac9365a82697cd5d#egg=colossalai

2
applications/Chat/requirements.txt

@ -2,7 +2,7 @@ transformers>=4.20.1
tqdm
datasets
loralib
colossalai==0.3.1
git+https://github.com/hpcaitech/ColossalAI.git@ef4b99ebcda823c32e221383ac9365a82697cd5d#egg=colossalai
torch<2.0.0, >=1.12.1
langchain
tokenizers

Loading…
Cancel
Save