Save Hugging Face Model with One Bin
作者:XD / 发表: 2023年8月7日 02:41 / 编程笔记/ 阅读量:840
Save Hugging Face Model with One Bin
max_shard_size (int or str, optional, defaults to "10GB") — Only applicable for models. The maximum size for a checkpoint before being sharded. Checkpoints shard will then be each of size lower than this size. If expressed as a string, needs to be digits followed by a unit (like "5MB").