mirror of https://github.com/hpcaitech/ColossalAI
b893342f95
* [fx] test tracer on diffuser modules. * [fx] shorter seq_len. * Update requirements-test.txt |
||
---|---|---|
.. | ||
test_hf_model | ||
test_timm_model | ||
test_torchaudio_model | ||
test_torchrec_model | ||
test_torchvision_model | ||
test_activation_checkpoint_annotation.py | ||
test_control_flow.py | ||
test_functional_conv.py | ||
test_patched_module.py | ||
test_patched_op.py |