shareit commited on
Commit
187f695
·
verified ·
1 Parent(s): dd32fa3

Initial upload: base + CS-chatbot LoRA + supervisor LoRA merged

Browse files
README.md ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: microsoft/Phi-4-reasoning
5
+ tags:
6
+ - customer-service
7
+ - supervisor
8
+ - cycleinstruct
9
+ - lg-electronics
10
+ - phi
11
+ - fine-tuned
12
+ language:
13
+ - ko
14
+ - en
15
+ - de
16
+ - fr
17
+ - es
18
+ pipeline_tag: text-generation
19
+ ---
20
+
21
+ # cycleinstruct-phi4-supervisor
22
+
23
+ Fully merged **microsoft/Phi-4-reasoning** (14.66 B) fine-tuned in two
24
+ stages for the LG-Electronics customer-service **quality-supervisor** task.
25
+ Given a `(Category, Conversation Transcript, Retrieved Document)` triplet,
26
+ the model emits
27
+
28
+ ```
29
+ <think>
30
+ [Query-Document Alignment] …
31
+ [Response-Document Consistency] …
32
+ [Response Completeness] …
33
+ </think>
34
+ {"label": "correct" | "incorrect", "reason": "…"}
35
+ ```
36
+
37
+ This repo contains a **single-file, ready-to-use** checkpoint — no adapter
38
+ merging required at load time.
39
+
40
+ ## Training pipeline (CycleInstruct-motivated, two-stage SFT)
41
+
42
+ Following the [CycleInstruct paper](https://arxiv.org/abs/2508.09551)
43
+ (EMNLP 2025) as the augmentation strategy motivator:
44
+
45
+ 1. **Stage 1 — CS-chatbot SFT** on 9,868 natural `(question, answer)`
46
+ pairs built from LG feedback + general-inquiry data. LoRA r=16 α=32,
47
+ Muon @ lr=2e-3, seed=1337, 8 epochs.
48
+ 2. **Stage 2 — Supervisor SFT** on 3,771 human-annotated supervisor
49
+ judgements. Stage-1 LoRA is merged into the base first, then a fresh
50
+ LoRA r=16 α=32 is added and trained with Muon @ lr=1e-3, seed=42,
51
+ 7 epochs on 4,096-token sequences.
52
+
53
+ The uploaded checkpoint is the result of merging **both** LoRA stages into
54
+ the base weights and re-saving with `save_pretrained`.
55
+
56
+ ## Metrics — 199-item held-out supervisor test set (T=0, `max_new_tokens=1200`)
57
+
58
+ | Metric | Stage-1 only | **This model (full merged)** |
59
+ |---|---|---|
60
+ | Parse-fail rate | 95.98 % | **0.00 %** |
61
+ | Accuracy | 1.01 % | **68.84 %** |
62
+ | Macro-F1 | 0.033 | **0.615** |
63
+ | chrF | 6.55 | **40.92** |
64
+ | ROUGE-L | 0.062 | **0.885** |
65
+ | BLEU-4 | 0.37 | **22.41** |
66
+ | BERTScore-F1 | 0.826 | **0.901** |
67
+ | SBERT-cos (multi-mpnet) | 0.437 | **0.830** |
68
+
69
+ Per-class:
70
+
71
+ | Class | Precision | Recall | F1 | Support |
72
+ |---|---|---|---|---|
73
+ | correct | 0.417 | 0.481 | 0.446 | 52 |
74
+ | incorrect | 0.806 | 0.762 | 0.783 | 147 |
75
+
76
+ ## Loading
77
+
78
+ ```python
79
+ from transformers import AutoTokenizer, AutoModelForCausalLM
80
+ import torch
81
+
82
+ REPO = "shareit/cycleinstruct-phi4-supervisor"
83
+
84
+ tok = AutoTokenizer.from_pretrained(REPO)
85
+ model = AutoModelForCausalLM.from_pretrained(
86
+ REPO, torch_dtype=torch.bfloat16,
87
+ attn_implementation="sdpa", device_map="auto").eval()
88
+
89
+ SYSTEM = "당신은 전자제품 CS 챗봇의 품질을 평가하는 수퍼바이저입니다."
90
+ USER = "[Category] W/M\n[Conversation Transcript] …\n[Retrieved Document] …"
91
+
92
+ # Phi-4-reasoning ChatML with our clean system prompt (skip default Thought scaffold)
93
+ prompt = (
94
+ f"<|im_start|>system<|im_sep|>{SYSTEM}<|im_end|>"
95
+ f"<|im_start|>user<|im_sep|>{USER}<|im_end|>"
96
+ f"<|im_start|>assistant<|im_sep|>"
97
+ )
98
+ out = model.generate(
99
+ **tok(prompt, return_tensors="pt", add_special_tokens=False).to(model.device),
100
+ do_sample=False, max_new_tokens=1200,
101
+ pad_token_id=tok.pad_token_id,
102
+ )
103
+ print(tok.decode(out[0], skip_special_tokens=False))
104
+ ```
105
+
106
+ `max_new_tokens=1200` matters — the `<think>` block usually consumes
107
+ 500-900 tokens before the final JSON verdict.
108
+
109
+ ## Training details (stage 2, on top of stage-1-merged base)
110
+
111
+ - **PEFT**: LoRA r=16, α=32, dropout 0.05, `target_modules=all-linear`, bias='none'
112
+ - **Optimizer**: Muon on 2D matrices (Newton-Schulz orthogonalisation) + AdamW on 1D params
113
+ - **LR**: 1e-3 (matrix) / 1e-4 (aux), cosine decay with 3 % warmup, grad-clip 1.0
114
+ - **Batch**: per-device 1 × grad-accum 16 (effective 16)
115
+ - **Seq len**: 4096 (user text char-clipped if exceeds; assistant always preserved)
116
+ - **Seed**: 42, **Epochs**: 7
117
+ - **Attention**: SDPA (bf16 native on H200)
118
+ - **Wall clock**: 5h48m on a half-H200 (48 GB active)
119
+
120
+ ## Data
121
+
122
+ - Stage-1 train: 9,868 `(q, a)` pairs from `data/processed/train_pairs.jsonl`
123
+ (multilingual, mostly English, ~50 % English, ~15 % German, then FR/ES/IT/JA/ZH…)
124
+ - Stage-2 train: 3,771 supervisor-annotated rows
125
+ `{"conversations": [{"from":"system", …}, {"from":"user", …}, {"from":"assistant", …}]}`
126
+ with the assistant response being a `<think>…</think>{"label":…,"reason":…}` judgement.
127
+ - Test: 199 held-out supervisor rows (unseen during either stage).
128
+
129
+ ## Intended use / limitations
130
+
131
+ - Intended for research reproduction of CycleInstruct-style continuation
132
+ training on labeled downstream tasks.
133
+ - The `correct` class has substantially lower F1 (0.446) than `incorrect`
134
+ (0.783), reflecting the 39/61 % class imbalance in the training data.
135
+ Class-weighted loss or balanced sampling would likely help.
136
+ - The `<think>` reasoning is Korean; input transcripts may be any language.
137
+
138
+ ## License
139
+
140
+ MIT (inherits from the `microsoft/Phi-4-reasoning` base model).
chat_template.jinja ADDED
@@ -0,0 +1 @@
 
 
1
+ <|im_start|>system<|im_sep|>You are Phi, a language model trained by Microsoft to help users. Your role as an assistant involves thoroughly exploring questions through a systematic thinking process before providing the final precise and accurate solutions. This requires engaging in a comprehensive cycle of analysis, summarizing, exploration, reassessment, reflection, backtracing, and iteration to develop well-considered thinking process. Please structure your response into two main sections: Thought and Solution using the specified format: <think> {Thought section} </think> {Solution section}. In the Thought section, detail your reasoning process in steps. Each step should include detailed considerations such as analysing questions, summarizing relevant findings, brainstorming new ideas, verifying the accuracy of the current steps, refining any errors, and revisiting previous steps. In the Solution section, based on various attempts, explorations, and reflections from the Thought section, systematically present the final solution that you deem correct. The Solution section should be logical, accurate, and concise and detail necessary steps needed to reach the conclusion. Now, try to solve the following question through the above guidelines:<|im_end|>{% for message in messages %}{% if (message['role'] == 'user') %}{{'<|im_start|>user<|im_sep|>' + message['content'] + '<|im_end|>'}}{% elif (message['role'] == 'assistant') %}{{'<|im_start|>assistant<|im_sep|>'}}{% generation %}{{message['content'] + '<|im_end|>'}}{% endgeneration %}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant<|im_sep|>' }}{% endif %}
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Phi3ForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 100257,
8
+ "dtype": "bfloat16",
9
+ "embd_pdrop": 0.0,
10
+ "eos_token_id": 100265,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 5120,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 17920,
15
+ "max_position_embeddings": 32768,
16
+ "model_type": "phi3",
17
+ "num_attention_heads": 40,
18
+ "num_hidden_layers": 40,
19
+ "num_key_value_heads": 10,
20
+ "original_max_position_embeddings": 32768,
21
+ "pad_token_id": 100349,
22
+ "partial_rotary_factor": 1.0,
23
+ "resid_pdrop": 0.0,
24
+ "rms_norm_eps": 1e-05,
25
+ "rope_parameters": {
26
+ "partial_rotary_factor": 1.0,
27
+ "rope_theta": 500000,
28
+ "rope_type": "default"
29
+ },
30
+ "sliding_window": null,
31
+ "tie_word_embeddings": false,
32
+ "transformers_version": "5.12.1",
33
+ "use_cache": true,
34
+ "vocab_size": 100352
35
+ }
generation_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 100257,
4
+ "do_sample": true,
5
+ "eos_token_id": 100265,
6
+ "pad_token_id": 100349,
7
+ "temperature": 0.8,
8
+ "top_k": 50,
9
+ "top_p": 0.95,
10
+ "transformers_version": "5.12.1"
11
+ }
model-00001-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:688e3f5e94051936e54a789e5683b567b8c0663a19f78d53eec6b4e1645b2908
3
+ size 3968923696
model-00002-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:294d5d5a4c92ccb9f79288ff212293342e428fd260060335d110d726d7566185
3
+ size 3722561416
model-00003-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2dce113162389f7a3593d032eaca63cd58c4f67a4018999841f11152a17540bf
3
+ size 3906072600
model-00004-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba8a77cc1a9d753b061f42e43a11e54ed112f20044cf162a75596ee6ae15afb5
3
+ size 3958490944
model-00005-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:678b294053ba8bc90690203f9bde3052b0d6224d6a34d6bae737a0291a4316d9
3
+ size 3722561448
model-00006-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47abfb5db3d59683d27ad7bfd734a0ae641468b2507ff588b8c7c523c11a9bd1
3
+ size 3906072624
model-00007-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:867284d5a00fa3b26f6d2988970ee93762f5792219af1d0a2caa4b5cae8b2432
3
+ size 3958490944
model-00008-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8808010d5843d3bfc2a4f7d9f0a110ae179cd6add557321bebf03e9a73a79a4f
3
+ size 2175869336
model.safetensors.index.json ADDED
@@ -0,0 +1,251 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_parameters": 14659507200,
4
+ "total_size": 29319014400
5
+ },
6
+ "weight_map": {
7
+ "lm_head.weight": "model-00001-of-00008.safetensors",
8
+ "model.embed_tokens.weight": "model-00001-of-00008.safetensors",
9
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00008.safetensors",
10
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00008.safetensors",
11
+ "model.layers.0.mlp.gate_up_proj.weight": "model-00001-of-00008.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00008.safetensors",
13
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00008.safetensors",
14
+ "model.layers.0.self_attn.qkv_proj.weight": "model-00001-of-00008.safetensors",
15
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00008.safetensors",
16
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00008.safetensors",
17
+ "model.layers.1.mlp.gate_up_proj.weight": "model-00001-of-00008.safetensors",
18
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00008.safetensors",
19
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00008.safetensors",
20
+ "model.layers.1.self_attn.qkv_proj.weight": "model-00001-of-00008.safetensors",
21
+ "model.layers.10.input_layernorm.weight": "model-00003-of-00008.safetensors",
22
+ "model.layers.10.mlp.down_proj.weight": "model-00003-of-00008.safetensors",
23
+ "model.layers.10.mlp.gate_up_proj.weight": "model-00003-of-00008.safetensors",
24
+ "model.layers.10.post_attention_layernorm.weight": "model-00003-of-00008.safetensors",
25
+ "model.layers.10.self_attn.o_proj.weight": "model-00003-of-00008.safetensors",
26
+ "model.layers.10.self_attn.qkv_proj.weight": "model-00003-of-00008.safetensors",
27
+ "model.layers.11.input_layernorm.weight": "model-00003-of-00008.safetensors",
28
+ "model.layers.11.mlp.down_proj.weight": "model-00003-of-00008.safetensors",
29
+ "model.layers.11.mlp.gate_up_proj.weight": "model-00003-of-00008.safetensors",
30
+ "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00008.safetensors",
31
+ "model.layers.11.self_attn.o_proj.weight": "model-00003-of-00008.safetensors",
32
+ "model.layers.11.self_attn.qkv_proj.weight": "model-00003-of-00008.safetensors",
33
+ "model.layers.12.input_layernorm.weight": "model-00003-of-00008.safetensors",
34
+ "model.layers.12.mlp.down_proj.weight": "model-00003-of-00008.safetensors",
35
+ "model.layers.12.mlp.gate_up_proj.weight": "model-00003-of-00008.safetensors",
36
+ "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00008.safetensors",
37
+ "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00008.safetensors",
38
+ "model.layers.12.self_attn.qkv_proj.weight": "model-00003-of-00008.safetensors",
39
+ "model.layers.13.input_layernorm.weight": "model-00003-of-00008.safetensors",
40
+ "model.layers.13.mlp.down_proj.weight": "model-00003-of-00008.safetensors",
41
+ "model.layers.13.mlp.gate_up_proj.weight": "model-00003-of-00008.safetensors",
42
+ "model.layers.13.post_attention_layernorm.weight": "model-00003-of-00008.safetensors",
43
+ "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00008.safetensors",
44
+ "model.layers.13.self_attn.qkv_proj.weight": "model-00003-of-00008.safetensors",
45
+ "model.layers.14.input_layernorm.weight": "model-00003-of-00008.safetensors",
46
+ "model.layers.14.mlp.down_proj.weight": "model-00004-of-00008.safetensors",
47
+ "model.layers.14.mlp.gate_up_proj.weight": "model-00004-of-00008.safetensors",
48
+ "model.layers.14.post_attention_layernorm.weight": "model-00004-of-00008.safetensors",
49
+ "model.layers.14.self_attn.o_proj.weight": "model-00004-of-00008.safetensors",
50
+ "model.layers.14.self_attn.qkv_proj.weight": "model-00004-of-00008.safetensors",
51
+ "model.layers.15.input_layernorm.weight": "model-00004-of-00008.safetensors",
52
+ "model.layers.15.mlp.down_proj.weight": "model-00004-of-00008.safetensors",
53
+ "model.layers.15.mlp.gate_up_proj.weight": "model-00004-of-00008.safetensors",
54
+ "model.layers.15.post_attention_layernorm.weight": "model-00004-of-00008.safetensors",
55
+ "model.layers.15.self_attn.o_proj.weight": "model-00004-of-00008.safetensors",
56
+ "model.layers.15.self_attn.qkv_proj.weight": "model-00004-of-00008.safetensors",
57
+ "model.layers.16.input_layernorm.weight": "model-00004-of-00008.safetensors",
58
+ "model.layers.16.mlp.down_proj.weight": "model-00004-of-00008.safetensors",
59
+ "model.layers.16.mlp.gate_up_proj.weight": "model-00004-of-00008.safetensors",
60
+ "model.layers.16.post_attention_layernorm.weight": "model-00004-of-00008.safetensors",
61
+ "model.layers.16.self_attn.o_proj.weight": "model-00004-of-00008.safetensors",
62
+ "model.layers.16.self_attn.qkv_proj.weight": "model-00004-of-00008.safetensors",
63
+ "model.layers.17.input_layernorm.weight": "model-00004-of-00008.safetensors",
64
+ "model.layers.17.mlp.down_proj.weight": "model-00004-of-00008.safetensors",
65
+ "model.layers.17.mlp.gate_up_proj.weight": "model-00004-of-00008.safetensors",
66
+ "model.layers.17.post_attention_layernorm.weight": "model-00004-of-00008.safetensors",
67
+ "model.layers.17.self_attn.o_proj.weight": "model-00004-of-00008.safetensors",
68
+ "model.layers.17.self_attn.qkv_proj.weight": "model-00004-of-00008.safetensors",
69
+ "model.layers.18.input_layernorm.weight": "model-00004-of-00008.safetensors",
70
+ "model.layers.18.mlp.down_proj.weight": "model-00004-of-00008.safetensors",
71
+ "model.layers.18.mlp.gate_up_proj.weight": "model-00004-of-00008.safetensors",
72
+ "model.layers.18.post_attention_layernorm.weight": "model-00004-of-00008.safetensors",
73
+ "model.layers.18.self_attn.o_proj.weight": "model-00004-of-00008.safetensors",
74
+ "model.layers.18.self_attn.qkv_proj.weight": "model-00004-of-00008.safetensors",
75
+ "model.layers.19.input_layernorm.weight": "model-00004-of-00008.safetensors",
76
+ "model.layers.19.mlp.down_proj.weight": "model-00004-of-00008.safetensors",
77
+ "model.layers.19.mlp.gate_up_proj.weight": "model-00004-of-00008.safetensors",
78
+ "model.layers.19.post_attention_layernorm.weight": "model-00004-of-00008.safetensors",
79
+ "model.layers.19.self_attn.o_proj.weight": "model-00005-of-00008.safetensors",
80
+ "model.layers.19.self_attn.qkv_proj.weight": "model-00005-of-00008.safetensors",
81
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00008.safetensors",
82
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00008.safetensors",
83
+ "model.layers.2.mlp.gate_up_proj.weight": "model-00001-of-00008.safetensors",
84
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00008.safetensors",
85
+ "model.layers.2.self_attn.o_proj.weight": "model-00002-of-00008.safetensors",
86
+ "model.layers.2.self_attn.qkv_proj.weight": "model-00002-of-00008.safetensors",
87
+ "model.layers.20.input_layernorm.weight": "model-00005-of-00008.safetensors",
88
+ "model.layers.20.mlp.down_proj.weight": "model-00005-of-00008.safetensors",
89
+ "model.layers.20.mlp.gate_up_proj.weight": "model-00005-of-00008.safetensors",
90
+ "model.layers.20.post_attention_layernorm.weight": "model-00005-of-00008.safetensors",
91
+ "model.layers.20.self_attn.o_proj.weight": "model-00005-of-00008.safetensors",
92
+ "model.layers.20.self_attn.qkv_proj.weight": "model-00005-of-00008.safetensors",
93
+ "model.layers.21.input_layernorm.weight": "model-00005-of-00008.safetensors",
94
+ "model.layers.21.mlp.down_proj.weight": "model-00005-of-00008.safetensors",
95
+ "model.layers.21.mlp.gate_up_proj.weight": "model-00005-of-00008.safetensors",
96
+ "model.layers.21.post_attention_layernorm.weight": "model-00005-of-00008.safetensors",
97
+ "model.layers.21.self_attn.o_proj.weight": "model-00005-of-00008.safetensors",
98
+ "model.layers.21.self_attn.qkv_proj.weight": "model-00005-of-00008.safetensors",
99
+ "model.layers.22.input_layernorm.weight": "model-00005-of-00008.safetensors",
100
+ "model.layers.22.mlp.down_proj.weight": "model-00005-of-00008.safetensors",
101
+ "model.layers.22.mlp.gate_up_proj.weight": "model-00005-of-00008.safetensors",
102
+ "model.layers.22.post_attention_layernorm.weight": "model-00005-of-00008.safetensors",
103
+ "model.layers.22.self_attn.o_proj.weight": "model-00005-of-00008.safetensors",
104
+ "model.layers.22.self_attn.qkv_proj.weight": "model-00005-of-00008.safetensors",
105
+ "model.layers.23.input_layernorm.weight": "model-00005-of-00008.safetensors",
106
+ "model.layers.23.mlp.down_proj.weight": "model-00005-of-00008.safetensors",
107
+ "model.layers.23.mlp.gate_up_proj.weight": "model-00005-of-00008.safetensors",
108
+ "model.layers.23.post_attention_layernorm.weight": "model-00005-of-00008.safetensors",
109
+ "model.layers.23.self_attn.o_proj.weight": "model-00005-of-00008.safetensors",
110
+ "model.layers.23.self_attn.qkv_proj.weight": "model-00005-of-00008.safetensors",
111
+ "model.layers.24.input_layernorm.weight": "model-00005-of-00008.safetensors",
112
+ "model.layers.24.mlp.down_proj.weight": "model-00005-of-00008.safetensors",
113
+ "model.layers.24.mlp.gate_up_proj.weight": "model-00005-of-00008.safetensors",
114
+ "model.layers.24.post_attention_layernorm.weight": "model-00005-of-00008.safetensors",
115
+ "model.layers.24.self_attn.o_proj.weight": "model-00005-of-00008.safetensors",
116
+ "model.layers.24.self_attn.qkv_proj.weight": "model-00005-of-00008.safetensors",
117
+ "model.layers.25.input_layernorm.weight": "model-00005-of-00008.safetensors",
118
+ "model.layers.25.mlp.down_proj.weight": "model-00005-of-00008.safetensors",
119
+ "model.layers.25.mlp.gate_up_proj.weight": "model-00006-of-00008.safetensors",
120
+ "model.layers.25.post_attention_layernorm.weight": "model-00006-of-00008.safetensors",
121
+ "model.layers.25.self_attn.o_proj.weight": "model-00006-of-00008.safetensors",
122
+ "model.layers.25.self_attn.qkv_proj.weight": "model-00006-of-00008.safetensors",
123
+ "model.layers.26.input_layernorm.weight": "model-00006-of-00008.safetensors",
124
+ "model.layers.26.mlp.down_proj.weight": "model-00006-of-00008.safetensors",
125
+ "model.layers.26.mlp.gate_up_proj.weight": "model-00006-of-00008.safetensors",
126
+ "model.layers.26.post_attention_layernorm.weight": "model-00006-of-00008.safetensors",
127
+ "model.layers.26.self_attn.o_proj.weight": "model-00006-of-00008.safetensors",
128
+ "model.layers.26.self_attn.qkv_proj.weight": "model-00006-of-00008.safetensors",
129
+ "model.layers.27.input_layernorm.weight": "model-00006-of-00008.safetensors",
130
+ "model.layers.27.mlp.down_proj.weight": "model-00006-of-00008.safetensors",
131
+ "model.layers.27.mlp.gate_up_proj.weight": "model-00006-of-00008.safetensors",
132
+ "model.layers.27.post_attention_layernorm.weight": "model-00006-of-00008.safetensors",
133
+ "model.layers.27.self_attn.o_proj.weight": "model-00006-of-00008.safetensors",
134
+ "model.layers.27.self_attn.qkv_proj.weight": "model-00006-of-00008.safetensors",
135
+ "model.layers.28.input_layernorm.weight": "model-00006-of-00008.safetensors",
136
+ "model.layers.28.mlp.down_proj.weight": "model-00006-of-00008.safetensors",
137
+ "model.layers.28.mlp.gate_up_proj.weight": "model-00006-of-00008.safetensors",
138
+ "model.layers.28.post_attention_layernorm.weight": "model-00006-of-00008.safetensors",
139
+ "model.layers.28.self_attn.o_proj.weight": "model-00006-of-00008.safetensors",
140
+ "model.layers.28.self_attn.qkv_proj.weight": "model-00006-of-00008.safetensors",
141
+ "model.layers.29.input_layernorm.weight": "model-00006-of-00008.safetensors",
142
+ "model.layers.29.mlp.down_proj.weight": "model-00006-of-00008.safetensors",
143
+ "model.layers.29.mlp.gate_up_proj.weight": "model-00006-of-00008.safetensors",
144
+ "model.layers.29.post_attention_layernorm.weight": "model-00006-of-00008.safetensors",
145
+ "model.layers.29.self_attn.o_proj.weight": "model-00006-of-00008.safetensors",
146
+ "model.layers.29.self_attn.qkv_proj.weight": "model-00006-of-00008.safetensors",
147
+ "model.layers.3.input_layernorm.weight": "model-00002-of-00008.safetensors",
148
+ "model.layers.3.mlp.down_proj.weight": "model-00002-of-00008.safetensors",
149
+ "model.layers.3.mlp.gate_up_proj.weight": "model-00002-of-00008.safetensors",
150
+ "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00008.safetensors",
151
+ "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00008.safetensors",
152
+ "model.layers.3.self_attn.qkv_proj.weight": "model-00002-of-00008.safetensors",
153
+ "model.layers.30.input_layernorm.weight": "model-00006-of-00008.safetensors",
154
+ "model.layers.30.mlp.down_proj.weight": "model-00006-of-00008.safetensors",
155
+ "model.layers.30.mlp.gate_up_proj.weight": "model-00006-of-00008.safetensors",
156
+ "model.layers.30.post_attention_layernorm.weight": "model-00006-of-00008.safetensors",
157
+ "model.layers.30.self_attn.o_proj.weight": "model-00006-of-00008.safetensors",
158
+ "model.layers.30.self_attn.qkv_proj.weight": "model-00006-of-00008.safetensors",
159
+ "model.layers.31.input_layernorm.weight": "model-00006-of-00008.safetensors",
160
+ "model.layers.31.mlp.down_proj.weight": "model-00007-of-00008.safetensors",
161
+ "model.layers.31.mlp.gate_up_proj.weight": "model-00007-of-00008.safetensors",
162
+ "model.layers.31.post_attention_layernorm.weight": "model-00007-of-00008.safetensors",
163
+ "model.layers.31.self_attn.o_proj.weight": "model-00007-of-00008.safetensors",
164
+ "model.layers.31.self_attn.qkv_proj.weight": "model-00007-of-00008.safetensors",
165
+ "model.layers.32.input_layernorm.weight": "model-00007-of-00008.safetensors",
166
+ "model.layers.32.mlp.down_proj.weight": "model-00007-of-00008.safetensors",
167
+ "model.layers.32.mlp.gate_up_proj.weight": "model-00007-of-00008.safetensors",
168
+ "model.layers.32.post_attention_layernorm.weight": "model-00007-of-00008.safetensors",
169
+ "model.layers.32.self_attn.o_proj.weight": "model-00007-of-00008.safetensors",
170
+ "model.layers.32.self_attn.qkv_proj.weight": "model-00007-of-00008.safetensors",
171
+ "model.layers.33.input_layernorm.weight": "model-00007-of-00008.safetensors",
172
+ "model.layers.33.mlp.down_proj.weight": "model-00007-of-00008.safetensors",
173
+ "model.layers.33.mlp.gate_up_proj.weight": "model-00007-of-00008.safetensors",
174
+ "model.layers.33.post_attention_layernorm.weight": "model-00007-of-00008.safetensors",
175
+ "model.layers.33.self_attn.o_proj.weight": "model-00007-of-00008.safetensors",
176
+ "model.layers.33.self_attn.qkv_proj.weight": "model-00007-of-00008.safetensors",
177
+ "model.layers.34.input_layernorm.weight": "model-00007-of-00008.safetensors",
178
+ "model.layers.34.mlp.down_proj.weight": "model-00007-of-00008.safetensors",
179
+ "model.layers.34.mlp.gate_up_proj.weight": "model-00007-of-00008.safetensors",
180
+ "model.layers.34.post_attention_layernorm.weight": "model-00007-of-00008.safetensors",
181
+ "model.layers.34.self_attn.o_proj.weight": "model-00007-of-00008.safetensors",
182
+ "model.layers.34.self_attn.qkv_proj.weight": "model-00007-of-00008.safetensors",
183
+ "model.layers.35.input_layernorm.weight": "model-00007-of-00008.safetensors",
184
+ "model.layers.35.mlp.down_proj.weight": "model-00007-of-00008.safetensors",
185
+ "model.layers.35.mlp.gate_up_proj.weight": "model-00007-of-00008.safetensors",
186
+ "model.layers.35.post_attention_layernorm.weight": "model-00007-of-00008.safetensors",
187
+ "model.layers.35.self_attn.o_proj.weight": "model-00007-of-00008.safetensors",
188
+ "model.layers.35.self_attn.qkv_proj.weight": "model-00007-of-00008.safetensors",
189
+ "model.layers.36.input_layernorm.weight": "model-00007-of-00008.safetensors",
190
+ "model.layers.36.mlp.down_proj.weight": "model-00007-of-00008.safetensors",
191
+ "model.layers.36.mlp.gate_up_proj.weight": "model-00007-of-00008.safetensors",
192
+ "model.layers.36.post_attention_layernorm.weight": "model-00007-of-00008.safetensors",
193
+ "model.layers.36.self_attn.o_proj.weight": "model-00008-of-00008.safetensors",
194
+ "model.layers.36.self_attn.qkv_proj.weight": "model-00008-of-00008.safetensors",
195
+ "model.layers.37.input_layernorm.weight": "model-00008-of-00008.safetensors",
196
+ "model.layers.37.mlp.down_proj.weight": "model-00008-of-00008.safetensors",
197
+ "model.layers.37.mlp.gate_up_proj.weight": "model-00008-of-00008.safetensors",
198
+ "model.layers.37.post_attention_layernorm.weight": "model-00008-of-00008.safetensors",
199
+ "model.layers.37.self_attn.o_proj.weight": "model-00008-of-00008.safetensors",
200
+ "model.layers.37.self_attn.qkv_proj.weight": "model-00008-of-00008.safetensors",
201
+ "model.layers.38.input_layernorm.weight": "model-00008-of-00008.safetensors",
202
+ "model.layers.38.mlp.down_proj.weight": "model-00008-of-00008.safetensors",
203
+ "model.layers.38.mlp.gate_up_proj.weight": "model-00008-of-00008.safetensors",
204
+ "model.layers.38.post_attention_layernorm.weight": "model-00008-of-00008.safetensors",
205
+ "model.layers.38.self_attn.o_proj.weight": "model-00008-of-00008.safetensors",
206
+ "model.layers.38.self_attn.qkv_proj.weight": "model-00008-of-00008.safetensors",
207
+ "model.layers.39.input_layernorm.weight": "model-00008-of-00008.safetensors",
208
+ "model.layers.39.mlp.down_proj.weight": "model-00008-of-00008.safetensors",
209
+ "model.layers.39.mlp.gate_up_proj.weight": "model-00008-of-00008.safetensors",
210
+ "model.layers.39.post_attention_layernorm.weight": "model-00008-of-00008.safetensors",
211
+ "model.layers.39.self_attn.o_proj.weight": "model-00008-of-00008.safetensors",
212
+ "model.layers.39.self_attn.qkv_proj.weight": "model-00008-of-00008.safetensors",
213
+ "model.layers.4.input_layernorm.weight": "model-00002-of-00008.safetensors",
214
+ "model.layers.4.mlp.down_proj.weight": "model-00002-of-00008.safetensors",
215
+ "model.layers.4.mlp.gate_up_proj.weight": "model-00002-of-00008.safetensors",
216
+ "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00008.safetensors",
217
+ "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00008.safetensors",
218
+ "model.layers.4.self_attn.qkv_proj.weight": "model-00002-of-00008.safetensors",
219
+ "model.layers.5.input_layernorm.weight": "model-00002-of-00008.safetensors",
220
+ "model.layers.5.mlp.down_proj.weight": "model-00002-of-00008.safetensors",
221
+ "model.layers.5.mlp.gate_up_proj.weight": "model-00002-of-00008.safetensors",
222
+ "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00008.safetensors",
223
+ "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00008.safetensors",
224
+ "model.layers.5.self_attn.qkv_proj.weight": "model-00002-of-00008.safetensors",
225
+ "model.layers.6.input_layernorm.weight": "model-00002-of-00008.safetensors",
226
+ "model.layers.6.mlp.down_proj.weight": "model-00002-of-00008.safetensors",
227
+ "model.layers.6.mlp.gate_up_proj.weight": "model-00002-of-00008.safetensors",
228
+ "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00008.safetensors",
229
+ "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00008.safetensors",
230
+ "model.layers.6.self_attn.qkv_proj.weight": "model-00002-of-00008.safetensors",
231
+ "model.layers.7.input_layernorm.weight": "model-00002-of-00008.safetensors",
232
+ "model.layers.7.mlp.down_proj.weight": "model-00002-of-00008.safetensors",
233
+ "model.layers.7.mlp.gate_up_proj.weight": "model-00002-of-00008.safetensors",
234
+ "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00008.safetensors",
235
+ "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00008.safetensors",
236
+ "model.layers.7.self_attn.qkv_proj.weight": "model-00002-of-00008.safetensors",
237
+ "model.layers.8.input_layernorm.weight": "model-00002-of-00008.safetensors",
238
+ "model.layers.8.mlp.down_proj.weight": "model-00002-of-00008.safetensors",
239
+ "model.layers.8.mlp.gate_up_proj.weight": "model-00003-of-00008.safetensors",
240
+ "model.layers.8.post_attention_layernorm.weight": "model-00003-of-00008.safetensors",
241
+ "model.layers.8.self_attn.o_proj.weight": "model-00003-of-00008.safetensors",
242
+ "model.layers.8.self_attn.qkv_proj.weight": "model-00003-of-00008.safetensors",
243
+ "model.layers.9.input_layernorm.weight": "model-00003-of-00008.safetensors",
244
+ "model.layers.9.mlp.down_proj.weight": "model-00003-of-00008.safetensors",
245
+ "model.layers.9.mlp.gate_up_proj.weight": "model-00003-of-00008.safetensors",
246
+ "model.layers.9.post_attention_layernorm.weight": "model-00003-of-00008.safetensors",
247
+ "model.layers.9.self_attn.o_proj.weight": "model-00003-of-00008.safetensors",
248
+ "model.layers.9.self_attn.qkv_proj.weight": "model-00003-of-00008.safetensors",
249
+ "model.norm.weight": "model-00008-of-00008.safetensors"
250
+ }
251
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "is_local": true,
8
+ "local_files_only": false,
9
+ "model_max_length": 32768,
10
+ "pad_token": "<|dummy_85|>",
11
+ "padding_side": "left",
12
+ "tokenizer_class": "TokenizersBackend"
13
+ }