..
__init__.py
[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert ( #5088 )
2023-11-28 16:54:42 +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
[devops] remove post commit ci ( #5566 )
2024-04-08 15:09:40 +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
[shardformer] Sequence Parallelism Optimization ( #5533 )
2024-04-03 17:15:47 +08:00
gptj.py
[workflow] fixed oom tests ( #5275 )
2024-01-16 18:55:13 +08:00
llama.py
[shardformer] Sequence Parallelism Optimization ( #5533 )
2024-04-03 17:15:47 +08:00
mistral.py
[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert ( #5088 )
2023-11-28 16:54:42 +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