subatomicseer commited on
Commit
3cd8d8a
1 Parent(s): 99560ba

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +36 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "SpeechTextAny2Any"
4
+ ],
5
+ "bos_token_id": 1,
6
+ "complementary_masking": false,
7
+ "encoder_config": {
8
+ "attn_dropout": 0.0,
9
+ "attn_flash": true,
10
+ "conv_dropout": 0.0,
11
+ "conv_expansion_factor": 2,
12
+ "conv_kernel_size": 5,
13
+ "depth": 8,
14
+ "dim_head": 32,
15
+ "ff_dropout": 0.0,
16
+ "ff_mult": 4,
17
+ "heads": 16
18
+ },
19
+ "eos_token_id": 2,
20
+ "hidden_size": 512,
21
+ "label_smoothing": 0.0,
22
+ "model_type": "speech_text_any2any",
23
+ "num_speech_tokens": 1024,
24
+ "num_text_tokens": 256,
25
+ "pad_token_id": 0,
26
+ "special_tokens": {
27
+ "[MASK]": 3,
28
+ "[PAD]": 0,
29
+ "[SEP]": 4,
30
+ "[SPEECH]": 2,
31
+ "[TEXT]": 1
32
+ },
33
+ "torch_dtype": "bfloat16",
34
+ "transformers_version": "4.41.1",
35
+ "uniform_masking": true
36
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b639f961650c655d00f88a4fe8067ae811f1fef6b0af021768b21379e058501
3
+ size 112567435