mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-14 20:16:01 +00:00
15 lines
123 B
Plaintext
15 lines
123 B
Plaintext
numpy
|
|
tqdm
|
|
psutil
|
|
packaging
|
|
pre-commit
|
|
rich
|
|
click
|
|
fabric
|
|
contexttimer
|
|
ninja
|
|
torch>=1.11
|
|
safetensors
|
|
flash_attn>=2.0
|
|
einops
|