github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 14:47:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
45b08f08cb8581986e513ef9162d93a8c07fd250
ColossalAI/colossalai/lazy
History
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
..
__init__.py
[lazy] refactor lazy init (#3891)
2023-06-05 14:20:47 +08:00
lazy_init.py
[lazy] support init on cuda (#4269)
2023-07-19 16:43:01 +08:00