Safetensors
kimi_linear
ror HF Staff commited on
Commit
b501957
·
verified ·
1 Parent(s): 0a4fe50

Upload 6 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
chat_template.jinja ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {% macro render_content(msg) -%}
2
+ {%- set c = msg.get('content') -%}
3
+ {%- if c is string -%}
4
+ {{ c }}
5
+ {%- elif c is not none -%}
6
+ {% for content in c -%}
7
+ {% if content['type'] == 'image' or 'image' in content or 'image_url' in content -%}
8
+ <|media_start|>image<|media_content|><|media_pad|><|media_end|>
9
+ {% else -%}
10
+ {{ content['text'] }}
11
+ {%- endif -%}
12
+ {%- endfor -%}
13
+ {%- endif -%}
14
+ {%- endmacro %}
15
+
16
+
17
+ {%- if tools -%}
18
+ <|im_system|>tool_declare<|im_middle|>{{ tools | tojson(separators=(',', ':')) }}<|im_end|>
19
+ {%- endif -%}
20
+ {% for message in messages %}
21
+ {%- set role_name = message.get('name') or message['role'] -%}
22
+ {%- if message['role'] == 'user' -%}
23
+ <|im_user|>{{role_name}}<|im_middle|>
24
+ {%- elif message['role'] == 'assistant' -%}
25
+ <|im_assistant|>{{role_name}}<|im_middle|>
26
+ {%- else -%}
27
+ <|im_system|>{{role_name}}<|im_middle|>
28
+ {%- endif -%}
29
+
30
+ {%- if message['role'] == 'assistant' and message.get('tool_calls') -%}
31
+ {{render_content(message)}}<|tool_calls_section_begin|>
32
+ {%- for tool_call in message['tool_calls'] -%}
33
+ {%- set formatted_id = tool_call['id'] -%}
34
+ <|tool_call_begin|>{{ formatted_id }}<|tool_call_argument_begin|>{% if tool_call['function']['arguments'] is string %}{{ tool_call['function']['arguments'] }}{% else %}{{ tool_call['function']['arguments'] | tojson }}{% endif %}<|tool_call_end|>
35
+ {%- endfor -%}
36
+ <|tool_calls_section_end|>
37
+ {%- elif message['role'] == 'tool' -%}
38
+ {%- set tool_call_id = message.tool_call_id -%}
39
+ ## Return of {{ tool_call_id }}
40
+ {{render_content(message)}}
41
+ {%- elif message['content'] is not none -%}
42
+ {{render_content(message)}}
43
+ {%- endif -%}
44
+ <|im_end|>
45
+ {%- endfor -%}
46
+ {%- if add_generation_prompt -%}
47
+ <|im_assistant|>assistant<|im_middle|>
48
+ {%- endif -%}
config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "KimiLinearForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 163584,
8
+ "dtype": "bfloat16",
9
+ "eos_token_id": 163586,
10
+ "head_dim": 8,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 64,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 128,
15
+ "kv_lora_rank": 16,
16
+ "layer_types": [
17
+ "linear_attention",
18
+ "linear_attention",
19
+ "linear_attention",
20
+ "full_attention"
21
+ ],
22
+ "linear_conv_kernel_dim": 4,
23
+ "linear_head_dim": 16,
24
+ "linear_num_heads": 4,
25
+ "max_position_embeddings": 1024,
26
+ "mlp_layer_types": [
27
+ "dense",
28
+ "sparse",
29
+ "sparse",
30
+ "sparse"
31
+ ],
32
+ "model_type": "kimi_linear",
33
+ "moe_intermediate_size": 32,
34
+ "n_group": 1,
35
+ "n_routed_experts": 8,
36
+ "n_shared_experts": 1,
37
+ "norm_topk_prob": true,
38
+ "num_attention_heads": 4,
39
+ "num_experts_per_tok": 2,
40
+ "num_hidden_layers": 4,
41
+ "num_key_value_heads": 4,
42
+ "pad_token_id": 163839,
43
+ "pretraining_tp": 1,
44
+ "q_lora_rank": null,
45
+ "qk_head_dim": 24,
46
+ "qk_nope_head_dim": 16,
47
+ "qk_rope_head_dim": 8,
48
+ "rms_norm_eps": 1e-05,
49
+ "routed_scaling_factor": 2.446,
50
+ "tie_word_embeddings": false,
51
+ "topk_group": 1,
52
+ "transformers_version": "5.16.0.dev0",
53
+ "use_cache": true,
54
+ "v_head_dim": 16,
55
+ "vocab_size": 163840
56
+ }
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 163584,
4
+ "eos_token_id": 163586,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "pad_token_id": 163839,
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:45a05f47fabe2b2785a6923507b15f0a49445ab488db609b14f3941411ed8db1
3
+ size 42503160
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96894d7eb8bd9ce759df54963eb77bdbcf709e329a8007ed5f5cb402b9b1615b
3
+ size 19592396
tokenizer_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "auto_map": {
3
+ "AutoTokenizer": [
4
+ "tokenization_kimi.TikTokenTokenizer",
5
+ null
6
+ ]
7
+ },
8
+ "backend": "tokenizers",
9
+ "bos_token": "[BOS]",
10
+ "clean_up_tokenization_spaces": false,
11
+ "eos_token": "[EOS]",
12
+ "is_local": false,
13
+ "local_files_only": false,
14
+ "model_max_length": 1000000000000000019884624838656,
15
+ "pad_token": "[PAD]",
16
+ "tokenizer_class": "TokenizersBackend",
17
+ "unk_token": "[UNK]"
18
+ }