将 safe.tensor 转换为 pytorch bin 文件

Question

我使用 HuggingFace 微调了我的 Transformer 模型。它为我提供了一个

model.safetensor

文件供以后使用。

我想将模型插入到仅需要

pytorch.bin

文件的旧框架中。我想知道如何降级我的模型以适应框架。

我不确定进行此类 Moodle 格式转换的最佳方法是什么。

感谢您的帮助。

Answer 1

如果使用以下代码保存模型，您将获得 .safetensors 格式的模型： model.save_pretrained('文件夹/').

使用以下代码保存模型，您将获得.bin格式的模型： torch.save(model.state_dict(),'folder/pytorch_model.bin'.format(epoch)).

或者，您可以使用 model.save_pretrained(output_dir, safe_serialization=False).

当您降级 Transformers 库时，它会自动将模型保存在 .bin 文件中。