hf-transformers-bot commited on
Commit
92014d5
·
verified ·
1 Parent(s): f292631

Update tiny models for PeAudioModel

Browse files
config.json ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "PeAudioModel"
4
+ ],
5
+ "audio_config": {
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "dac_config": {
9
+ "_name_or_path": "",
10
+ "architectures": null,
11
+ "chunk_size_feed_forward": 0,
12
+ "codebook_dim": 32,
13
+ "codebook_loss_weight": 1.0,
14
+ "codebook_size": 512,
15
+ "commitment_loss_weight": 0.25,
16
+ "decoder_hidden_size": 16,
17
+ "downsampling_ratios": [
18
+ 2,
19
+ 4,
20
+ 4
21
+ ],
22
+ "dtype": null,
23
+ "encoder_hidden_size": 16,
24
+ "hidden_size": 128,
25
+ "hop_length": 32,
26
+ "id2label": {
27
+ "0": "LABEL_0",
28
+ "1": "LABEL_1"
29
+ },
30
+ "is_encoder_decoder": false,
31
+ "label2id": {
32
+ "LABEL_0": 0,
33
+ "LABEL_1": 1
34
+ },
35
+ "model_type": "dac",
36
+ "n_codebooks": 6,
37
+ "output_attentions": false,
38
+ "output_hidden_states": false,
39
+ "problem_type": null,
40
+ "quantizer_dropout": 0.0,
41
+ "return_dict": true,
42
+ "sampling_rate": 16000,
43
+ "upsampling_ratios": [
44
+ 4,
45
+ 4,
46
+ 2
47
+ ]
48
+ },
49
+ "head_dim": 128,
50
+ "hidden_act": "silu",
51
+ "hidden_size": 32,
52
+ "initializer_range": 0.02,
53
+ "intermediate_size": 37,
54
+ "max_position_embeddings": 512,
55
+ "max_window_layers": 28,
56
+ "model_type": "pe_audio_encoder",
57
+ "num_attention_heads": 2,
58
+ "num_hidden_layers": 2,
59
+ "num_key_value_heads": 2,
60
+ "rms_norm_eps": 1e-05,
61
+ "rope_parameters": {
62
+ "rope_theta": 20000,
63
+ "rope_type": "default"
64
+ }
65
+ },
66
+ "dtype": "float32",
67
+ "model_type": "pe_audio",
68
+ "projection_dim": 32,
69
+ "text_config": {
70
+ "attention_bias": false,
71
+ "attention_dropout": 0.0,
72
+ "bos_token_id": 50281,
73
+ "classifier_activation": "gelu",
74
+ "classifier_bias": false,
75
+ "classifier_dropout": 0.0,
76
+ "classifier_pooling": "cls",
77
+ "cls_token_id": 50281,
78
+ "decoder_bias": true,
79
+ "deterministic_flash_attn": false,
80
+ "embedding_dropout": 0.0,
81
+ "eos_token_id": 50282,
82
+ "hidden_activation": "gelu",
83
+ "hidden_size": 32,
84
+ "initializer_cutoff_factor": 2.0,
85
+ "initializer_range": 0.02,
86
+ "intermediate_size": 37,
87
+ "is_decoder": false,
88
+ "layer_types": [
89
+ "full_attention",
90
+ "sliding_attention"
91
+ ],
92
+ "local_attention": 128,
93
+ "max_position_embeddings": 512,
94
+ "mlp_bias": false,
95
+ "mlp_dropout": 0.0,
96
+ "model_type": "modernbert",
97
+ "norm_bias": false,
98
+ "norm_eps": 1e-05,
99
+ "num_attention_heads": 4,
100
+ "num_hidden_layers": 2,
101
+ "pad_token_id": 50283,
102
+ "rope_parameters": {
103
+ "full_attention": {
104
+ "rope_theta": 160000.0,
105
+ "rope_type": "default"
106
+ },
107
+ "sliding_attention": {
108
+ "rope_theta": 10000.0,
109
+ "rope_type": "default"
110
+ }
111
+ },
112
+ "sep_token_id": 50282,
113
+ "sparse_pred_ignore_index": -100,
114
+ "sparse_prediction": false,
115
+ "tie_word_embeddings": true,
116
+ "type_vocab_size": 16,
117
+ "vocab_size": 50368
118
+ },
119
+ "transformers_version": "5.16.0.dev0"
120
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5928a9122bfcdefcce4da2202b4cb782b2716063b28003223e5b942f934e072
3
+ size 7933096
preprocessor_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "feature_extractor_type": "PeAudioFeatureExtractor",
3
+ "feature_size": 1,
4
+ "hop_length": 1920,
5
+ "padding_side": "right",
6
+ "padding_value": 0.0,
7
+ "return_attention_mask": true,
8
+ "sampling_rate": 48000
9
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "clean_up_tokenization_spaces": true,
4
+ "cls_token": "[CLS]",
5
+ "extra_special_tokens": [],
6
+ "is_local": true,
7
+ "local_files_only": false,
8
+ "mask_token": "[MASK]",
9
+ "model_input_names": [
10
+ "input_ids",
11
+ "attention_mask"
12
+ ],
13
+ "model_max_length": 8192,
14
+ "pad_token": "[PAD]",
15
+ "processor_class": "PeAudioVideoProcessor",
16
+ "sep_token": "[SEP]",
17
+ "tokenizer_class": "TokenizersBackend",
18
+ "unk_token": "[UNK]"
19
+ }
video_preprocessor_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_format": "channels_first",
3
+ "default_to_square": true,
4
+ "do_convert_rgb": true,
5
+ "do_normalize": true,
6
+ "do_rescale": true,
7
+ "do_resize": true,
8
+ "do_sample_frames": false,
9
+ "image_mean": [
10
+ 0.5,
11
+ 0.5,
12
+ 0.5
13
+ ],
14
+ "image_std": [
15
+ 0.5,
16
+ 0.5,
17
+ 0.5
18
+ ],
19
+ "resample": 2,
20
+ "rescale_factor": 0.00392156862745098,
21
+ "return_metadata": false,
22
+ "size": {
23
+ "height": 336,
24
+ "width": 336
25
+ },
26
+ "video_processor_type": "PeVideoVideoProcessor"
27
+ }