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/booster
Baizhou Zhang 38ccb8b1a3
[shardformer] support from_pretrained when loading model with HybridParallelPlugin (#4575)
1 year ago
..
mixed_precision
plugin [shardformer] support from_pretrained when loading model with HybridParallelPlugin (#4575) 1 year ago
__init__.py
accelerator.py
booster.py [misc] resolve code factor issues (#4433) 1 year ago