Jiarui Fang f38006ea83 [checkpoint] checkpoint for ColoTensor Model (#1196)

2022-07-06 17:22:03 +08:00

..

__init__.py

fix import error (#880)

2022-04-26 19:28:40 +08:00

colo_init_context.py

[checkpoint] checkpoint for ColoTensor Model (#1196)

2022-07-06 17:22:03 +08:00

lazy_init_context.py

[context]support arbitary module materialization. (#1193)

2022-07-04 10:12:02 +08:00

utils.py

[context]use meta tensor to init model lazily. (#1187)

2022-06-29 21:02:30 +08:00