github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-24 13:03:00 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
hotfix/example_test
ColossalAI/tests/test_lazy/test_models.py
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

964 B
Raw Permalink Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink