ColossalAI/tests/test_lazy
..
lazy_init_utils.py
test_from_pretrained.py
test_models.py
test_ops.py