remove convert.py

pull/4101/head
zxcd 2 weeks ago
parent 86f2bd1f6a
commit 6869f4cd6f

@ -1,18 +0,0 @@
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')

@ -75,7 +75,7 @@ class WhisperExecutor(BaseExecutor):
'--size', '--size',
type=str, type=str,
default='large', default='large',
choices=['large', 'medium', 'base', 'small', 'tiny'], choices=['large', 'medium', 'base', 'small', 'tiny', 'turbo'],
help='Choose model size. now only support large, large:[whisper-large-16k]' help='Choose model size. now only support large, large:[whisper-large-16k]'
) )
self.parser.add_argument( self.parser.add_argument(

Loading…
Cancel
Save