Blame · requirements/requirements.txt · 0d0a5820331768896a7bd743cf6407586dd534ca · OpenDAS / ColossalAI · GitLab

Find file Normal view History Permalink

requirements.txt 172 Bytes

Newer Older

Migrated project zbian committed Oct 28, 2021	1 2 3	`numpy tqdm psutil`
Update GitHub action and pre-commit settings (#196) BoxiangW committed Jan 28, 2022	4 5	`packaging pre-commit`
[log] better logging display with rich (#426) Jiarui Fang committed Mar 16, 2022	6	`rich`
[cli] add missing requirement (#805) Frank Lee committed Apr 19, 2022	7	`click`
[CLI] refactored the launch CLI and fixed bugs in multi-node launching (#844) Frank Lee committed Apr 24, 2022	8	`fabric`
[setup] supported conda-installed torch (#2048) Frank Lee committed Nov 30, 2022	9	`contexttimer`
[buider] use builder() for cpu adam and fused optim in setup.py (#2187) Jiarui Fang committed Dec 23, 2022	10	`ninja`
[zero] support multiple (partial) backward passes (#5596) Hongxin Liu committed Apr 16, 2024	11	`torch>=2.1.0`
[checkpoint] refactored the API and added safetensors support (#3427) Frank Lee committed Apr 04, 2023	12	`safetensors`
[doc] add Series A Funding and NeurIPS news (#4377) binmakeswell committed Aug 04, 2023	13	`einops`
[Inference] Dynamic Batching Inference, online and offline (#4953) Jianghai committed Oct 30, 2023	14 15	`pydantic ray`
[Refactor] Integrated some lightllm kernels into token-attention (#4946) Cuiqing Li committed Oct 19, 2023	16 17 18	`sentencepiece google protobuf`
[shardformer] update transformers (#5583) Wang Binluo committed Apr 24, 2024	19	`transformers==4.36.2`