hf-transformers-bot HF Staff commited on
Commit
2f59f42
·
verified ·
1 Parent(s): 5366d9e

Update tiny models for GraniteSpeech5ForCTC

Browse files
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "GraniteSpeech5ForCTC"
4
+ ],
5
+ "ctc_loss_reduction": "mean",
6
+ "ctc_zero_infinity": true,
7
+ "dtype": "float32",
8
+ "encoder_config": {
9
+ "activation_dropout": 0.0,
10
+ "attention_bias": true,
11
+ "attention_dropout": 0.0,
12
+ "context_size": 13,
13
+ "conv_expansion_factor": 2,
14
+ "conv_kernel_size": 7,
15
+ "head_dim": 16,
16
+ "hidden_act": "silu",
17
+ "hidden_size": 32,
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 64,
20
+ "max_position_embeddings": 64,
21
+ "model_type": "granite_speech5_encoder",
22
+ "num_attention_heads": 2,
23
+ "num_hidden_layers": 2,
24
+ "num_key_value_heads": 2,
25
+ "num_mel_bins": 6,
26
+ "subsample_layers": [
27
+ 0
28
+ ],
29
+ "vocab_size": 16384
30
+ },
31
+ "initializer_range": 0.02,
32
+ "model_type": "granite_speech5_ctc",
33
+ "pad_token_id": 0,
34
+ "tie_word_embeddings": true,
35
+ "transformers_version": "5.19.0.dev0",
36
+ "vocab_size": 16384
37
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "output_attentions": false,
4
+ "output_hidden_states": false,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "5.19.0.dev0"
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f51f28612bd15c5826f7bfc4cfad614c25d5a014d952dcb0d775361a0afa3d97
3
+ size 4445936
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "is_local": true,
4
+ "local_files_only": false,
5
+ "model_max_length": 1000000000000000019884624838656,
6
+ "pad_token": "<|blank|>",
7
+ "padding": true,
8
+ "processor_class": "GraniteSpeech5Processor",
9
+ "tokenizer_class": "ParakeetTokenizer"
10
+ }