1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-13 19:07:33 +00:00
ColossalAI/requirements/requirements.txt
Frank Lee 1beb85cc25
[checkpoint] refactored the API and added safetensors support ()
* [checkpoint] refactored the API and added safetensors support

* polish code
2023-04-04 15:23:01 +08:00

13 lines
100 B
Plaintext

numpy
tqdm
psutil
packaging
pre-commit
rich
click
fabric
contexttimer
ninja
torch>=1.11
safetensors