How to shard a model

Is there an easy way to obtain the sharded-fp16 version of a model after having loaded it with from_pretrained?

1 Like