ColossalAI/tests/kit/model_zoo/transformers
Hongxin Liu b8e770c832
[test] merge old components to test to model zoo (#4945)
* [test] add custom models in model zoo

* [test] update legacy test

* [test] update model zoo

* [test] update gemini test

* [test] remove components to test
2023-10-20 10:35:08 +08:00
..
__init__.py rename chatglm to chatglm2 (#4484) 2023-08-22 14:13:31 +08:00
albert.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
bert.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
blip2.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
bloom.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
chatglm2.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
gpt.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
llama.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
opt.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
sam.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
t5.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
vit.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
whisper.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00