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/policies
flybird11111 20190b49a5
[shardformer] to fix whisper test failed due to significant accuracy differences. (#4710)
1 year ago
..
__init__.py
auto_policy.py
base_policy.py
bert.py
blip2.py
bloom.py
chatglm2.py
gpt2.py
llama.py
opt.py
sam.py
t5.py
vit.py
whisper.py