mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-19 00:16:51 +00:00
be400a0936
* [chat] fix gemini strategy * [chat] fix gemini strategy * [chat] fix gemini strategy * [chat] fix gemini strategy * g# This is a combination of 2 commits. [chat] fix gemini strategy fox * [chat] fix gemini strategy update llama2 example [chat] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * [fix] fix gemini strategy * fix * fix * fix * fix * fix * Update train_prompts.py
15 lines
166 B
Plaintext
15 lines
166 B
Plaintext
transformers>=4.20.1
|
|
tqdm
|
|
datasets
|
|
loralib
|
|
colossalai==0.3.3
|
|
torch<2.0.0, >=1.12.1
|
|
langchain
|
|
tokenizers
|
|
fastapi
|
|
sse_starlette
|
|
wandb
|
|
sentencepiece
|
|
gpustat
|
|
tensorboard
|