botbw 80c3c8789b [Test/CI] remove test cases to reduce CI duration (#5753) ... * [test] smaller gpt2 test case * [test] reduce test cases: tests/test_zero/test_gemini/test_zeroddp_state_dict.py * [test] reduce test cases: tests/test_zero/test_gemini/test_grad_accum.py * [test] reduce test cases tests/test_zero/test_gemini/test_optim.py * Revert "[test] smaller gpt2 test case" Some tests might depend on the size of model (num of chunks) This reverts commit df705a5210. * [test] reduce test cases: tests/test_checkpoint_io/test_gemini_checkpoint_io.py * [CI] smaller test model for two mwo the two modifid cases * [CI] hardcode gpt model for tests/test_zero/test_gemini/test_search.py since we need a fixed answer there		2024-06-05 11:29:04 +08:00
..
__init__.py	[Shardformer] Support the Qwen2 model (#5699)	2024-05-09 20:04:25 +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	[shardformer] fix chatglm implementation (#5644)	2024-04-25 14:41:17 +08:00
falcon.py	[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088)	2023-11-28 16:54:42 +08:00
gpt.py	[Test/CI] remove test cases to reduce CI duration (#5753)	2024-06-05 11:29:04 +08:00
gptj.py	[workflow] fixed oom tests (#5275)	2024-01-16 18:55:13 +08:00
llama.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
mistral.py	[shardformer] update transformers (#5583)	2024-04-24 22:51:50 +08:00
opt.py	[test] merge old components to test to model zoo (#4945)	2023-10-20 10:35:08 +08:00
qwen2.py	[Shardformer] Support the Qwen2 model (#5699)	2024-05-09 20:04:25 +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