github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-17 16:08:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b366f1d99fd77dac73403b15eb590144e3bc2fdf
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