github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/requirements/requirements.txt

13 lines
100 B
Plaintext
Raw Normal View History

Migrated project
2021-10-28 16:21:23 +00:00
numpy
tqdm
psutil
Update GitHub action and pre-commit settings (#196) * Update GitHub action and pre-commit settings * Update GitHub action and pre-commit settings (#198)
2022-01-28 08:59:53 +00:00
packaging
pre-commit
[log] better logging display with rich (#426) * better logger using rich * remove deepspeed in zero requirements
2022-03-16 01:51:15 +00:00
rich
[cli] add missing requirement (#805)
2022-04-19 05:56:59 +00:00
click
[CLI] refactored the launch CLI and fixed bugs in multi-node launching (#844) * [cli] fixed multi-node job launching * [cli] fixed a bug in version comparison * [cli] support launching with env var * [cli] fixed multi-node job launching * [cli] fixed a bug in version comparison * [cli] support launching with env var * added docstring * [cli] added extra launch arguments * [cli] added default launch rdzv args * [cli] fixed version comparison * [cli] added docstring examples and requierment * polish docstring * polish code * polish code
2022-04-24 05:26:26 +00:00
fabric
[setup] supported conda-installed torch (#2048) * [setup] supported conda-installed torch * polish code
2022-11-30 08:45:15 +00:00
contexttimer
[buider] use builder() for cpu adam and fused optim in setup.py (#2187)
2022-12-23 08:05:13 +00:00
ninja
fix torch 2.0 compatibility (#3346)
2023-03-30 07:25:24 +00:00
torch>=1.11
[checkpoint] refactored the API and added safetensors support (#3427) * [checkpoint] refactored the API and added safetensors support * polish code
2023-04-04 07:23:01 +00:00
safetensors