GuangyaoZhang 363cde6957 merge model and attention forward		2024-06-18 02:32:41 +00:00
..
chatglm2_6b	[hotfix] fix typo change enabel to enable under colossalai/shardformer/ (#5317)	2024-03-05 21:48:46 +08:00
__init__.py	…
bert.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	2024-04-29 15:33:51 +08:00
blip2.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	2024-04-29 15:33:51 +08:00
bloom.py	[shardformer] fix modeling of bloom and falcon (#5796)	2024-06-11 17:43:50 +08:00
chatglm2.py	…
command.py	merge model and attention forward	2024-06-18 02:32:41 +00:00
falcon.py	[shardformer] fix modeling of bloom and falcon (#5796)	2024-06-11 17:43:50 +08:00
gpt2.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	2024-06-14 10:59:33 +08:00
gptj.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	2024-06-14 10:59:33 +08:00
jit.py	…
llama.py	Support 4d parallel + flash attention (#5789)	2024-06-17 17:40:47 +08:00
mistral.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	2024-06-14 10:59:33 +08:00
opt.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	2024-05-21 11:07:13 +08:00
qwen2.py	[shardformer] fix import (#5788)	2024-06-06 19:09:50 +08:00
sam.py	…
t5.py	…
vit.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	2024-04-29 15:33:51 +08:00
whisper.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	2024-06-14 10:59:33 +08:00