add convert.py

pull/4101/head
zxcd 1 month ago
parent 4c83b49a8d
commit 8a1d5085f9

@ -0,0 +1,18 @@
import paddle
import torch
torch_model_dict = torch.load('large-v3-turbo.pt')['model_state_dict']
paddle_model_state_dict = {}
for key, val in torch_model_dict.items():
if key.endswith(
'weight'
) and val.ndim == 2 and key != "decoder.token_embedding.weight":
val = val.T
paddle_model_state_dict[key] = paddle.to_tensor(
val.cpu().numpy()).astype("float32")
# add other params in case if need, such as:
paddle_model_state_dict['dims'] = torch.load('large-v3-turbo.pt')['dims']
paddle.save(paddle_model_state_dict, 'weights.params')
Loading…
Cancel
Save