Files
ColossalAI/requirements/requirements.txt
Frank Lee 1beb85cc25 [checkpoint] refactored the API and added safetensors support (#3427)
* [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