mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-26 09:42:27 +00:00
2ddf624a86
* [shardformer]upgrade transformers for gpt2/gptj/whisper (#5807) * [shardformer] fix modeling of gpt2 and gptj * [shardformer] fix whisper modeling * [misc] update requirements --------- Co-authored-by: ver217 <lhx0217@gmail.com> * [shardformer]upgrade transformers for mistral (#5808) * upgrade transformers for mistral * fix * fix * [shardformer]upgrade transformers for llama (#5809) * update transformers fix * fix * fix * [inference] upgrade transformers (#5810) * update transformers fix * fix * fix * fix * fix * [gemini] update transformers for gemini (#5814) --------- Co-authored-by: ver217 <lhx0217@gmail.com>
26 lines
261 B
Plaintext
26 lines
261 B
Plaintext
numpy
|
|
tqdm
|
|
psutil
|
|
packaging
|
|
pre-commit
|
|
rich
|
|
click
|
|
fabric
|
|
contexttimer
|
|
ninja
|
|
torch>=2.1.0,<2.3.0
|
|
safetensors
|
|
einops
|
|
pydantic
|
|
ray
|
|
sentencepiece
|
|
google
|
|
protobuf
|
|
transformers==4.39.3
|
|
peft>=0.7.1
|
|
bitsandbytes>=0.39.0
|
|
rpyc==6.0.0
|
|
fastapi
|
|
uvicorn==0.29.0
|
|
galore_torch
|