You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Hongxin Liu a39a5c66fe Merge branch 'main' into feature/shardformer		1 year ago
..
__init__.py	rename chatglm to chatglm2 (#4484)	1 year ago
albert.py	[gemini] improve compatibility and add static placement policy (#4479)	1 year ago
bert.py	[gemini] improve compatibility and add static placement policy (#4479)	1 year ago
blip2.py	[Shardformer] Merge flash attention branch to pipeline branch (#4362)	1 year ago
bloom.py	[shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395)	1 year ago
chatglm2.py	[shardformer] chatglm support sequence parallel (#4482)	1 year ago
gpt.py	[gemini] improve compatibility and add static placement policy (#4479)	1 year ago
llama.py	[shardformer] adapted T5 and LLaMa test to use kit (#4049)	1 year ago
opt.py	[test] Hotfix/fix some model test and refactor check util api (#4369)	1 year ago
sam.py	[shardformer] support SAM (#4231)	1 year ago
t5.py	[shardformer]update t5 tests for using all optimizations. (#4407)	1 year ago
vit.py	[shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395)	1 year ago
whisper.py	[Shardformer] Merge flash attention branch to pipeline branch (#4362)	1 year ago