ColossalAI/requirements
Hongxin Liu fc5cef2c79
[lazy] support init on cuda (#4269)
* [lazy] support init on cuda

* [test] update lazy init test

* [test] fix transformer version
2023-07-19 16:43:01 +08:00
..
requirements-test.txt [lazy] support init on cuda (#4269) 2023-07-19 16:43:01 +08:00
requirements.txt [checkpoint] refactored the API and added safetensors support (#3427) 2023-04-04 15:23:01 +08:00