flybird11111 c7d6975d29 [shardformer] fix GPT2DoubleHeadsModel (#4703)		2023-09-13 15:57:16 +08:00
..
chatglm2_6b	[pipeline] add chatglm (#4363)	2023-08-15 23:25:14 +08:00
__init__.py	…
bert.py	[shardformer] vit/llama/t5 ignore the sequence parallelism flag and some fix. (#4498)	2023-08-24 15:50:02 +08:00
blip2.py	[shardformer] update shardformer to use flash attention 2 (#4392)	2023-08-15 23:25:14 +08:00
bloom.py	[shardformer] bloom support sequence parallel (#4465)	2023-08-18 15:34:18 +08:00
chatglm2.py	[shardformer] chatglm support sequence parallel (#4482)	2023-08-22 23:59:31 +08:00
gpt2.py	[shardformer] fix GPT2DoubleHeadsModel (#4703)	2023-09-13 15:57:16 +08:00
jit.py	[Shardformer] Merge flash attention branch to pipeline branch (#4362)	2023-08-15 23:25:14 +08:00
llama.py	[Feature] The first PR to Add TP inference engine, kv-cache manager and related kernels for our inference system (#4577)	2023-09-12 01:22:56 +08:00
opt.py	[shardformer] update llama2/opt finetune example and fix llama2 policy (#4645)	2023-09-09 22:45:36 +08:00
sam.py	[Shardformer] Merge flash attention branch to pipeline branch (#4362)	2023-08-15 23:25:14 +08:00
t5.py	[misc] resolve code factor issues (#4433)	2023-08-15 23:25:14 +08:00
vit.py	[example] update vit example for hybrid parallel plugin (#4641)	2023-09-07 17:38:45 +08:00
whisper.py	[shardformer] Pipeline/whisper (#4456)	2023-08-18 21:29:25 +08:00