From 8a1d5085f903b40796ce72e71d94c79da3c7f837 Mon Sep 17 00:00:00 2001 From: zxcd <228587199@qq.com> Date: Fri, 25 Jul 2025 11:17:41 +0800 Subject: [PATCH] add convert.py --- convert_torch2paddle.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 convert_torch2paddle.py diff --git a/convert_torch2paddle.py b/convert_torch2paddle.py new file mode 100644 index 000000000..b13a793a8 --- /dev/null +++ b/convert_torch2paddle.py @@ -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')