将加载了lora权重的模型参数保存为pytorch.bin
时间: 2023-06-03 07:01:53 浏览: 718
基于pytorch的保存和加载模型参数的方法
Sure, 您可以使用以下代码将加载了Lora权重的模型参数保存为pytorch.bin:
```
import torch
model = torch.load('lora_model.pth')
torch.save(model.state_dict(), 'lora_pytorch.bin')
```
请确保已将模型参数加载到`model`变量中,并将保存的文件名更改为您需要的名称。
阅读全文