adding support to override the configured model shard