Senstella senstella
senstella
/ parakeet-nemo-to-mlx.py
Created
May 6, 2025 14:04
A simple script to convert NeMo Parakeet weights to MLX.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| from safetensors.torch import save_file | |
| INPUT_NAME = "model_weights.ckpt" | |
| OUTPUT_NAME = "model.safetensors" | |
| state = torch.load(INPUT_NAME, map_location="cpu") | |
| new_state = {} | |
| for key, value in state.items(): |