parakeet : verify hparams loaded from parakeet model bin file (#3950)

* verify hparams loaded from parakeet model bin file

* flexible way to accommodate CI as well security concern & test case addition.

* add bad model for CI tests

* removing whitespaces,couple of nits
This commit is contained in:
Bhargav Krish
2026-07-30 06:59:23 +02:00
committed by GitHub
parent a630b35c6f
commit 4523d0ce37
6 changed files with 142 additions and 18 deletions
Binary file not shown.
+10 -3
View File
@@ -3,6 +3,7 @@ import struct
import sys
import numpy as np
from pathlib import Path
import argparse
def write_tensor(fout, name, data):
n_dims = len(data.shape)
@@ -16,7 +17,7 @@ def write_tensor(fout, name, data):
fout.write(name_bytes)
data.tofile(fout)
def generate(output_path):
def generate(output_path, n_fft_override=None):
rng = np.random.default_rng(42)
hparams = {
@@ -37,6 +38,9 @@ def generate(output_path):
'n_max_tokens': 5,
}
if n_fft_override is not None:
hparams['n_fft'] = n_fft_override
n_vocab = hparams['n_vocab']
n_state = hparams['n_audio_state']
n_head = hparams['n_audio_head']
@@ -178,5 +182,8 @@ def generate(output_path):
print(f"Generated {output_path} ({size / 1024:.1f} KB)")
if __name__ == '__main__':
output = sys.argv[1] if len(sys.argv) > 1 else 'models/for-tests-ggml-parakeet-tdt.bin'
generate(output)
parser = argparse.ArgumentParser()
parser.add_argument('output', nargs='?', default='models/for-tests-ggml-parakeet-tdt.bin')
parser.add_argument('--n-fft',type=int, default=None)
args = parser.parse_args()
generate(args.output, args.n_fft)