Caitlin Blackmore
add model shards to reduce memory consumption
88a5ae5