ColossalAI/tests/components_to_test/__init__.py

from . import (
    bert,
    gpt2,
    hanging_param_model,
    inline_op_model,
    nested_model,
    repeated_computed_layer,
    resnet,
    simple_net,
)
from .utils import run_fwd_bwd

from . import albert    # isort:skip
[test] align model name with the file name. (#2045) 2 years ago			`from . import (`
			`bert,`
			`gpt2,`
			`hanging_param_model,`
			`inline_op_model,`
			`nested_model,`
			`repeated_computed_layer,`
			`resnet,`
			`simple_net,`
			`)`
[Gemini] add unitests to check gemini correctness (#2015) 2 years ago			`from .utils import run_fwd_bwd`
[Gemini] add albert in test models. (#2075) 2 years ago
			`from . import albert # isort:skip`