ColossalAI/tests/test_lazy
Frank Lee d69cd2eb89
[workflow] fixed oom tests (#5275)
* [workflow] fixed oom tests

* polish

* polish

* polish
2024-01-16 18:55:13 +08:00
..
lazy_init_utils.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_from_pretrained.py [lazy] support from_pretrained (#4801) 2023-09-26 11:04:11 +08:00
test_models.py [workflow] fixed oom tests (#5275) 2024-01-16 18:55:13 +08:00
test_ops.py [lazy] support torch 2.0 (#4763) 2023-09-21 16:30:23 +08:00