hf-transformers-bot commited on
Commit
b89d9d6
·
verified ·
1 Parent(s): ee0eba6

Update tiny models for RwkvForCausalLM

Browse files
config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu",
3
+ "architectures": [
4
+ "RwkvForCausalLM"
5
+ ],
6
+ "attention_hidden_size": 32,
7
+ "attn_pdrop": 0.1,
8
+ "bos_token_id": 0,
9
+ "context_length": 1024,
10
+ "dtype": "float32",
11
+ "eos_token_id": 0,
12
+ "gradient_checkpointing": false,
13
+ "hidden_size": 32,
14
+ "intermediate_size": 37,
15
+ "is_decoder": true,
16
+ "layer_norm_epsilon": 1e-05,
17
+ "model_type": "rwkv",
18
+ "n_positions": 512,
19
+ "num_hidden_layers": 2,
20
+ "pad_token_id": 1,
21
+ "reorder_and_upcast_attn": false,
22
+ "rescale_every": 6,
23
+ "resid_pdrop": 0.1,
24
+ "scale_attn_by_inverse_layer_idx": false,
25
+ "tie_word_embeddings": false,
26
+ "transformers_version": "5.16.0.dev0",
27
+ "type_vocab_size": 16,
28
+ "use_cache": true,
29
+ "vocab_size": 50277
30
+ }
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "pad_token_id": 1,
8
+ "transformers_version": "5.16.0.dev0",
9
+ "use_cache": true
10
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6934b0929ba74102e478d98646c28cf43db5f110ad2cb9dd8116157d12e2851
3
+ size 12938392
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": true,
6
+ "eos_token": "<|endoftext|>",
7
+ "errors": "replace",
8
+ "is_local": true,
9
+ "local_files_only": false,
10
+ "model_max_length": 1024,
11
+ "pad_token": "<|padding|>",
12
+ "tokenizer_class": "GPTNeoXTokenizer",
13
+ "trim_offsets": true,
14
+ "unk_token": "<|endoftext|>"
15
+ }