sanmonga22 commited on
Commit
ab13e3a
·
verified ·
1 Parent(s): 791c04e

Upload v81/whisper-small.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. v81/whisper-small.json +46 -0
v81/whisper-small.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "_comment": "Whisper-small ASR (encoder-decoder). Host log-mel + host conv stem (the conv is HTP-hostile in-graph) feed the encoder graph (input encoder_features); the autoregressive decoder (causal self-attention + cross-attention to the cached encoder states, learned positional embeddings, tied lm-head) emits tokens, detok'd on the host. Gated on WER vs HF. Same proven asr_transcribe host-op path as whisper-base (12 layers / d768 / 12 heads).",
4
+ "model": {
5
+ "name": "whisper-small",
6
+ "family": "asr",
7
+ "dsp_arch": "v81",
8
+ "tokenizer_pre": "whisper"
9
+ },
10
+ "params": {
11
+ "hidden": 768,
12
+ "vocab": 51865,
13
+ "n_layers": 12,
14
+ "max_ctx": 200,
15
+ "kv_dim": 768,
16
+ "head_dim": 64,
17
+ "rope_theta": 10000.0,
18
+ "eos_token_id": 50257
19
+ },
20
+ "artifacts": {
21
+ "contexts": {
22
+ "encoder": {
23
+ "bin": "whispersmall_enc_f16.bin"
24
+ },
25
+ "decoder": {
26
+ "bin": "whispersmall_dec_f16.bin"
27
+ }
28
+ },
29
+ "tokenizer": "tokenizer.json"
30
+ },
31
+ "plan": {
32
+ "steps": [
33
+ {
34
+ "host": "asr_transcribe",
35
+ "params": {
36
+ "encoder": "whispersmall_enc_f16",
37
+ "decoder": "whispersmall_dec_f16",
38
+ "sot": 50258,
39
+ "mel_filters": "whisper_small_mel_filters.bin",
40
+ "conv_weights": "whisper_conv_stem.bin",
41
+ "n_mels": 80
42
+ }
43
+ }
44
+ ]
45
+ }
46
+ }