pytorch 将拥抱脸模型转换为完整的.pt文件

idfiyjo8 于 12个月前发布在其他

关注(0)|答案(1)|浏览(151)

我已经下载了一个拥抱脸模型，它附带了各种文件，包括pytorch_model. bin，pytorch. json等。我的目标是将这个模型集成到我的项目中，我想将它转换为一个.pt文件以便于使用。有人可以指导我完成步骤或提供脚本来实现这一点吗？
其他信息：

Hugging Face Model Name: IndicWhisper

字符串
我已经探索了使用transformers库，但我不确定这种转换的具体步骤。任何帮助，指导，或示例脚本将不胜感激。谢谢！
我尝试将pytorch_model. bin重命名为pytorch_model.pt并使用它，但失败了。错误消息显示为KeyError：'dims'

1条答案

您可能正在查找这部分文档：https://huggingface.co/docs/transformers/v4.15.0/serialization#saving-a-model

此代码片段显示了如何使用TorchScript导出BertModel。在这里，BertModel根据BertConfig类示例化，然后以文件名traced_bert.pt保存到磁盘 *