Hugging Face push_to_hub为何转.bin而非safetensors?如何避免?
保留Safetensors格式推送模型到Hub的解决方法
要阻止模型被转换为PyTorch的.bin格式,只需在调用push_to_hub方法时添加safe_serialization=True参数即可,该参数会强制以safetensors格式完成序列化与推送。
修改后的代码如下:
model.push_to_hub('chukypedro/' + new_model, use_temp_dir=False, safe_serialization=True) tokenizer.push_to_hub('chukypedro/' + new_model, use_temp_dir=False)
注意:tokenizer的推送无需添加此参数,因为其保存格式不受该参数影响。
内容的提问来源于stack exchange,提问作者jojo okolie
相关产品推荐
相关产品推荐

