ColossalAI/colossalai/utils/model
Hongxin Liu 152239bbfa
[gemini] gemini supports lazy init (#3379)
* [gemini] fix nvme optimizer init

* [gemini] gemini supports lazy init

* [gemini] add init example

* [gemini] add fool model

* [zero] update gemini ddp

* [zero] update init example

* add chunk method

* add chunk method

* [lazyinit] fix lazy tensor tolist

* [gemini] fix buffer materialization

* [misc] remove useless file

* [booster] update gemini plugin

* [test] update gemini plugin test

* [test] fix gemini plugin test

* [gemini] fix import

* [gemini] fix import

* [lazyinit] use new metatensor

* [lazyinit] use new metatensor

* [lazyinit] fix __set__ method
2023-04-12 16:03:25 +08:00
..
__init__.py fix import error (#880) 2022-04-26 19:28:40 +08:00
experimental.py [gemini] gemini supports lazy init (#3379) 2023-04-12 16:03:25 +08:00
lazy_init_context.py [utils] fixed lazy init context (#1867) 2022-11-10 15:17:20 +08:00
utils.py [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00