ColossalAI/colossalai/lazy
Hongxin Liu 3e05c07bb8
[lazy] support torch 2.0 (#4763)
* [lazy] support _like methods and clamp

* [lazy] pass transformers models

* [lazy] fix device move and requires grad

* [lazy] fix requires grad and refactor api

* [lazy] fix requires grad
2023-09-21 16:30:23 +08:00
..
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
construction.py [lazy] support torch 2.0 (#4763) 2023-09-21 16:30:23 +08:00
lazy_init.py [lazy] support torch 2.0 (#4763) 2023-09-21 16:30:23 +08:00