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.
ColossalAI/colossalai/shardformer/modeling
GuangyaoZhang d84d68601a
change 'xxx if xxx else None' to 'xxx or None'
5 months ago
..
chatglm2_6b
__init__.py
bert.py [shardformer] support bias_gelu_jit_fused for models (#5647) 7 months ago
blip2.py [shardformer] support bias_gelu_jit_fused for models (#5647) 7 months ago
bloom.py [shardformer] fix modeling of bloom and falcon (#5796) 6 months ago
chatglm2.py [shardformer] fix chatglm implementation (#5644) 7 months ago
command.py change 'xxx if xxx else None' to 'xxx or None' 5 months ago
falcon.py [shardformer] fix modeling of bloom and falcon (#5796) 6 months ago
gpt2.py [shardformer] upgrade transformers to 4.39.3 (#5815) 6 months ago
gptj.py [shardformer] upgrade transformers to 4.39.3 (#5815) 6 months ago
jit.py
llama.py Support 4d parallel + flash attention (#5789) 5 months ago
mistral.py [shardformer] upgrade transformers to 4.39.3 (#5815) 6 months ago
opt.py [Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702) 6 months ago
qwen2.py [shardformer] fix import (#5788) 6 months ago
sam.py
t5.py
vit.py [shardformer] support bias_gelu_jit_fused for models (#5647) 7 months ago
whisper.py [shardformer] upgrade transformers to 4.39.3 (#5815) 6 months ago