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
flybird11111 6af6d6fc9f
[shardformer] support bias_gelu_jit_fused for models (#5647)
7 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
chatglm2.py
falcon.py
gpt2.py [shardformer] support bias_gelu_jit_fused for models (#5647) 7 months ago
gptj.py
jit.py
llama.py
mistral.py
opt.py
sam.py
t5.py
vit.py [shardformer] support bias_gelu_jit_fused for models (#5647) 7 months ago
whisper.py