github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_fx/test_tracer/test_hf_model
History
Super Daniel b893342f95
[fx] test tracer on diffuser modules. (#1750) 
* [fx] test tracer on diffuser modules.

* [fx] shorter seq_len.

* Update requirements-test.txt
 		2022-10-20 18:25:05 +08:00
..
test_hf_albert.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00
test_hf_bert.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00
test_hf_diffuser.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00
test_hf_gpt.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00
test_hf_opt.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00
test_hf_t5.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00
utils.py [fx] test tracer on diffuser modules. (#1750) 2022-10-20 18:25:05 +08:00