hf-transformers-bot commited on
Commit
ef7c2fc
·
verified ·
1 Parent(s): 443a5bf

Update tiny models for xLSTMModel

Browse files
Files changed (4) hide show
  1. config.json +42 -0
  2. model.safetensors +3 -0
  3. tokenizer.json +0 -0
  4. tokenizer_config.json +15 -0
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_out_norm": true,
3
+ "architectures": [
4
+ "xLSTMModel"
5
+ ],
6
+ "autocast_kernel_dtype": "bfloat16",
7
+ "bos_token_id": 0,
8
+ "chunk_size": 64,
9
+ "chunkwise_kernel": "chunkwise--native_autograd",
10
+ "dtype": "float32",
11
+ "embedding_dim": 128,
12
+ "eos_token_id": 0,
13
+ "eps": 1e-06,
14
+ "ffn_proj_factor": 2.667,
15
+ "ffn_round_up_to_multiple_of": 64,
16
+ "gate_soft_cap": 15.0,
17
+ "hidden_size": 128,
18
+ "inference_state_dtype": "float32",
19
+ "max_inference_chunksize": 16384,
20
+ "mode": "inference",
21
+ "model_type": "xlstm",
22
+ "n_positions": 512,
23
+ "norm_eps": 1e-06,
24
+ "norm_reduction_force_float32": true,
25
+ "num_blocks": 2,
26
+ "num_heads": 2,
27
+ "num_hidden_layers": 2,
28
+ "output_logit_soft_cap": 30.0,
29
+ "pad_token_id": 50276,
30
+ "qk_dim_factor": 0.5,
31
+ "return_last_states": true,
32
+ "sequence_kernel": "native_sequence__native",
33
+ "step_kernel": "native",
34
+ "tie_word_embeddings": false,
35
+ "transformers_version": "5.16.0.dev0",
36
+ "type_vocab_size": 16,
37
+ "use_bias": false,
38
+ "use_cache": true,
39
+ "v_dim_factor": 1.0,
40
+ "vocab_size": 50277,
41
+ "weight_mode": "single"
42
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ed137714dfae889135204311acd859faad47956015eaba310a1ee6428e522c7
3
+ size 27456688
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|endoftext|>",
7
+ "errors": "replace",
8
+ "is_local": true,
9
+ "local_files_only": false,
10
+ "model_max_length": 1000000000000000019884624838656,
11
+ "pad_token": null,
12
+ "tokenizer_class": "GPTNeoXTokenizer",
13
+ "trim_offsets": true,
14
+ "unk_token": "<|endoftext|>"
15
+ }