hf-transformers-bot commited on
Commit
618ae4d
·
verified ·
1 Parent(s): e22f5a8

Update tiny models for BarkModel

Browse files
config.json ADDED
@@ -0,0 +1,96 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BarkModel"
4
+ ],
5
+ "coarse_acoustics_config": {
6
+ "bias": true,
7
+ "block_size": 256,
8
+ "bos_token_id": 32,
9
+ "dropout": 0.0,
10
+ "eos_token_id": 32,
11
+ "hidden_size": 16,
12
+ "initializer_range": 0.02,
13
+ "input_vocab_size": 300,
14
+ "model_type": "coarse_acoustics",
15
+ "num_heads": 2,
16
+ "num_layers": 2,
17
+ "output_vocab_size": 300,
18
+ "pad_token_id": 32,
19
+ "use_cache": true
20
+ },
21
+ "codec_config": {
22
+ "audio_channels": 2,
23
+ "chunk_in_sec": null,
24
+ "chunk_length_s": null,
25
+ "codebook_dim": 32,
26
+ "codebook_size": 64,
27
+ "compress": 2,
28
+ "dilation_growth_rate": 2,
29
+ "hidden_size": 32,
30
+ "kernel_size": 7,
31
+ "last_kernel_size": 7,
32
+ "model_type": "encodec",
33
+ "norm_type": "weight_norm",
34
+ "normalize": false,
35
+ "num_filters": 8,
36
+ "num_lstm_layers": 1,
37
+ "num_residual_layers": 1,
38
+ "overlap": null,
39
+ "pad_mode": "reflect",
40
+ "residual_kernel_size": 3,
41
+ "sampling_rate": 24000,
42
+ "target_bandwidths": [
43
+ 1.5,
44
+ 3.0,
45
+ 6.0,
46
+ 12.0,
47
+ 24.0
48
+ ],
49
+ "trim_right_ratio": 1.0,
50
+ "upsampling_ratios": [
51
+ 8,
52
+ 4
53
+ ],
54
+ "use_causal_conv": true,
55
+ "use_conv_shortcut": true
56
+ },
57
+ "dtype": "float32",
58
+ "fine_acoustics_config": {
59
+ "bias": true,
60
+ "block_size": 256,
61
+ "bos_token_id": 32,
62
+ "dropout": 0.0,
63
+ "eos_token_id": 32,
64
+ "hidden_size": 16,
65
+ "initializer_range": 0.02,
66
+ "input_vocab_size": 300,
67
+ "model_type": "fine_acoustics",
68
+ "n_codes_given": 1,
69
+ "n_codes_total": 8,
70
+ "num_heads": 2,
71
+ "num_layers": 2,
72
+ "output_vocab_size": 300,
73
+ "pad_token_id": 32,
74
+ "tie_word_embeddings": true,
75
+ "use_cache": true
76
+ },
77
+ "initializer_range": 0.02,
78
+ "model_type": "bark",
79
+ "semantic_config": {
80
+ "bias": true,
81
+ "block_size": 256,
82
+ "bos_token_id": 32,
83
+ "dropout": 0.0,
84
+ "eos_token_id": 32,
85
+ "hidden_size": 16,
86
+ "initializer_range": 0.02,
87
+ "input_vocab_size": 300,
88
+ "model_type": "semantic",
89
+ "num_heads": 2,
90
+ "num_layers": 2,
91
+ "output_vocab_size": 300,
92
+ "pad_token_id": 32,
93
+ "use_cache": true
94
+ },
95
+ "transformers_version": "5.16.0.dev0"
96
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "output_attentions": false,
4
+ "output_hidden_states": false,
5
+ "transformers_version": "5.16.0.dev0"
6
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f2e3ecd388a3bc2993c52fbc09704031e2ec2395afb439059b488b6fe4a52ca
3
+ size 934372
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "clean_up_tokenization_spaces": true,
4
+ "cls_token": "[CLS]",
5
+ "do_lower_case": false,
6
+ "is_local": true,
7
+ "local_files_only": false,
8
+ "mask_token": "[MASK]",
9
+ "model_max_length": 512,
10
+ "pad_token": "[PAD]",
11
+ "processor_class": "BarkProcessor",
12
+ "sep_token": "[SEP]",
13
+ "strip_accents": null,
14
+ "tokenize_chinese_chars": true,
15
+ "tokenizer_class": "BertTokenizer",
16
+ "unk_token": "[UNK]"
17
+ }