ColossalAI/tests/test_fx/test_tracer/test_hf_model
YuliangLiu0306 19438ea0ef
[hotfix] skip gpt tracing test (#2064)
2022-12-02 16:48:28 +08:00
..
hf_tracer_utils.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_hf_albert.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_hf_bert.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_hf_diffuser.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_hf_gpt.py [hotfix] skip gpt tracing test (#2064) 2022-12-02 16:48:28 +08:00
test_hf_opt.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
test_hf_t5.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00