liumindmind commited on
Commit
56018c2
·
verified ·
1 Parent(s): bd6f46b

Upload folder using huggingface_hub

Browse files
args.json ADDED
@@ -0,0 +1,370 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "/data/leizikun/msp/script/occupy/output/minicpm5-1b-fullft-neko",
3
+ "per_device_train_batch_size": 4,
4
+ "num_train_epochs": 5.0,
5
+ "max_steps": -1,
6
+ "learning_rate": 2e-05,
7
+ "lr_scheduler_type": "cosine",
8
+ "lr_scheduler_kwargs": null,
9
+ "warmup_steps": 0,
10
+ "optim": "adamw_torch_fused",
11
+ "optim_args": null,
12
+ "weight_decay": 0.01,
13
+ "adam_beta1": 0.9,
14
+ "adam_beta2": 0.95,
15
+ "adam_epsilon": 1e-08,
16
+ "optim_target_modules": null,
17
+ "gradient_accumulation_steps": 4,
18
+ "average_tokens_across_devices": true,
19
+ "max_grad_norm": 1.0,
20
+ "label_smoothing_factor": 0.0,
21
+ "bf16": true,
22
+ "fp16": false,
23
+ "bf16_full_eval": false,
24
+ "fp16_full_eval": false,
25
+ "tf32": null,
26
+ "gradient_checkpointing": true,
27
+ "gradient_checkpointing_kwargs": null,
28
+ "torch_compile": false,
29
+ "torch_compile_backend": null,
30
+ "torch_compile_mode": null,
31
+ "use_liger_kernel": false,
32
+ "liger_kernel_config": null,
33
+ "use_cache": false,
34
+ "neftune_noise_alpha": null,
35
+ "torch_empty_cache_steps": null,
36
+ "auto_find_batch_size": false,
37
+ "logging_strategy": "steps",
38
+ "logging_steps": 10,
39
+ "logging_first_step": true,
40
+ "log_on_each_node": true,
41
+ "logging_nan_inf_filter": true,
42
+ "include_num_input_tokens_seen": false,
43
+ "log_level": "passive",
44
+ "log_level_replica": "warning",
45
+ "disable_tqdm": null,
46
+ "report_to": [
47
+ "wandb"
48
+ ],
49
+ "run_name": "minicpm5-1b-fullft-2a800-z2-bs4-ga4-4096",
50
+ "project": "huggingface",
51
+ "trackio_space_id": null,
52
+ "trackio_bucket_id": null,
53
+ "trackio_static_space_id": null,
54
+ "eval_strategy": "no",
55
+ "eval_steps": null,
56
+ "eval_delay": 0,
57
+ "per_device_eval_batch_size": 4,
58
+ "prediction_loss_only": false,
59
+ "eval_on_start": false,
60
+ "eval_do_concat_batches": true,
61
+ "eval_use_gather_object": false,
62
+ "eval_accumulation_steps": null,
63
+ "include_for_metrics": [],
64
+ "batch_eval_metrics": false,
65
+ "save_only_model": true,
66
+ "save_strategy": "epoch",
67
+ "save_steps": 500,
68
+ "save_on_each_node": false,
69
+ "save_total_limit": 3,
70
+ "enable_jit_checkpoint": false,
71
+ "push_to_hub": false,
72
+ "hub_token": null,
73
+ "hub_private_repo": null,
74
+ "hub_model_id": null,
75
+ "hub_strategy": "every_save",
76
+ "hub_always_push": false,
77
+ "hub_revision": null,
78
+ "load_best_model_at_end": false,
79
+ "metric_for_best_model": "loss",
80
+ "greater_is_better": false,
81
+ "ignore_data_skip": false,
82
+ "restore_callback_states_from_checkpoint": false,
83
+ "full_determinism": false,
84
+ "seed": 42,
85
+ "data_seed": 42,
86
+ "use_cpu": false,
87
+ "accelerator_config": {
88
+ "dispatch_batches": false
89
+ },
90
+ "parallelism_config": null,
91
+ "dataloader_drop_last": false,
92
+ "dataloader_num_workers": 4,
93
+ "dataloader_pin_memory": true,
94
+ "dataloader_persistent_workers": false,
95
+ "dataloader_prefetch_factor": null,
96
+ "remove_unused_columns": true,
97
+ "label_names": null,
98
+ "train_sampling_strategy": "random",
99
+ "length_column_name": "length",
100
+ "ddp_find_unused_parameters": false,
101
+ "ddp_bucket_cap_mb": null,
102
+ "ddp_broadcast_buffers": null,
103
+ "ddp_static_graph": null,
104
+ "ddp_backend": "nccl",
105
+ "ddp_timeout": 18000000,
106
+ "fsdp": [],
107
+ "fsdp_config": null,
108
+ "deepspeed": {
109
+ "bf16": {
110
+ "enabled": "auto"
111
+ },
112
+ "zero_optimization": {
113
+ "stage": 2,
114
+ "offload_optimizer": {
115
+ "device": "none",
116
+ "pin_memory": true
117
+ },
118
+ "allgather_partitions": true,
119
+ "allgather_bucket_size": 500000000.0,
120
+ "overlap_comm": true,
121
+ "reduce_scatter": true,
122
+ "reduce_bucket_size": 500000000.0,
123
+ "contiguous_gradients": true
124
+ },
125
+ "gradient_accumulation_steps": "auto",
126
+ "gradient_clipping": "auto",
127
+ "train_batch_size": "auto",
128
+ "train_micro_batch_size_per_gpu": "auto",
129
+ "wall_clock_breakdown": false
130
+ },
131
+ "debug": null,
132
+ "skip_memory_metrics": true,
133
+ "do_train": false,
134
+ "do_eval": false,
135
+ "do_predict": false,
136
+ "resume_from_checkpoint": null,
137
+ "warmup_ratio": 0.03,
138
+ "logging_dir": "/data/leizikun/msp/script/occupy/output/minicpm5-1b-fullft-neko/runs",
139
+ "local_rank": 0,
140
+ "sortish_sampler": false,
141
+ "predict_with_generate": false,
142
+ "generation_max_length": null,
143
+ "generation_num_beams": null,
144
+ "generation_config": null,
145
+ "tuner_backend": "peft",
146
+ "vit_gradient_checkpointing": false,
147
+ "router_aux_loss_coef": 0.0,
148
+ "enable_dft_loss": false,
149
+ "enable_channel_loss": false,
150
+ "safe_serialization": true,
151
+ "max_shard_size": "5GB",
152
+ "check_model": true,
153
+ "acc_strategy": "token",
154
+ "train_dataloader_shuffle": true,
155
+ "group_by_length": false,
156
+ "max_epochs": null,
157
+ "aligner_lr": null,
158
+ "vit_lr": null,
159
+ "use_logits_to_keep": null,
160
+ "ds3_gather_for_generation": true,
161
+ "resume_only_model": false,
162
+ "optimizer": null,
163
+ "loss_type": null,
164
+ "eval_metric": null,
165
+ "callbacks": [],
166
+ "early_stop_interval": null,
167
+ "eval_use_evalscope": false,
168
+ "eval_dataset": [],
169
+ "eval_dataset_args": null,
170
+ "eval_limit": null,
171
+ "eval_generation_config": null,
172
+ "extra_eval_args": null,
173
+ "tuner_type": "full",
174
+ "use_galore": false,
175
+ "galore_target_modules": null,
176
+ "galore_rank": 128,
177
+ "galore_update_proj_gap": 50,
178
+ "galore_scale": 1.0,
179
+ "galore_proj_type": "std",
180
+ "galore_optim_per_parameter": false,
181
+ "galore_with_embedding": false,
182
+ "galore_quantization": false,
183
+ "galore_proj_quant": false,
184
+ "galore_proj_bits": 4,
185
+ "galore_proj_group_size": 256,
186
+ "galore_cos_threshold": 0.4,
187
+ "galore_gamma_proj": 2,
188
+ "galore_queue_size": 5,
189
+ "lisa_activated_layers": 0,
190
+ "lisa_step_interval": 20,
191
+ "use_flash_ckpt": false,
192
+ "use_ray": false,
193
+ "ray_exp_name": null,
194
+ "device_groups": null,
195
+ "model": "/data/leizikun/msp/model/MiniCPM5-1B",
196
+ "model_type": "llama",
197
+ "model_revision": null,
198
+ "task_type": "causal_lm",
199
+ "torch_dtype": "bfloat16",
200
+ "attn_impl": "sdpa",
201
+ "experts_impl": null,
202
+ "new_special_tokens": [],
203
+ "num_labels": null,
204
+ "problem_type": null,
205
+ "rope_scaling": null,
206
+ "device_map": null,
207
+ "max_memory": {},
208
+ "max_model_len": null,
209
+ "local_repo_path": null,
210
+ "init_strategy": null,
211
+ "template": "minicpm5",
212
+ "system": null,
213
+ "max_length": 4096,
214
+ "truncation_strategy": "delete",
215
+ "max_pixels": null,
216
+ "agent_template": null,
217
+ "norm_bbox": null,
218
+ "use_chat_template": true,
219
+ "padding_side": "right",
220
+ "padding_free": false,
221
+ "loss_scale": "default",
222
+ "sequence_parallel_size": 1,
223
+ "is_binary_loss_scale": null,
224
+ "template_backend": "swift",
225
+ "response_prefix": null,
226
+ "enable_thinking": null,
227
+ "preserve_thinking": null,
228
+ "add_non_thinking_prefix": true,
229
+ "dataset": [
230
+ "/data/leizikun/msp/script/occupy/data/train_neko.jsonl"
231
+ ],
232
+ "val_dataset": [],
233
+ "cached_dataset": [],
234
+ "cached_val_dataset": [],
235
+ "split_dataset_ratio": 0.0,
236
+ "dataset_num_proc": 1,
237
+ "load_from_cache_file": false,
238
+ "dataset_shuffle": true,
239
+ "val_dataset_shuffle": false,
240
+ "streaming": false,
241
+ "interleave_prob": null,
242
+ "stopping_strategy": "first_exhausted",
243
+ "shuffle_buffer_size": 1000,
244
+ "download_mode": "reuse_dataset_if_exists",
245
+ "columns": {},
246
+ "strict": false,
247
+ "disable_auto_column_mapping": false,
248
+ "model_name": null,
249
+ "model_author": null,
250
+ "custom_dataset_info": [],
251
+ "quant_method": null,
252
+ "quant_bits": null,
253
+ "hqq_axis": null,
254
+ "bnb_4bit_compute_dtype": "bfloat16",
255
+ "bnb_4bit_quant_type": "nf4",
256
+ "bnb_4bit_use_double_quant": true,
257
+ "bnb_4bit_quant_storage": null,
258
+ "max_new_tokens": 64,
259
+ "temperature": 0.0,
260
+ "top_k": null,
261
+ "top_p": null,
262
+ "repetition_penalty": null,
263
+ "num_beams": 1,
264
+ "stream": false,
265
+ "stop_words": [],
266
+ "logprobs": false,
267
+ "top_logprobs": null,
268
+ "structured_outputs_regex": null,
269
+ "adapters": [],
270
+ "external_plugins": [],
271
+ "custom_register_path": [],
272
+ "model_kwargs": {},
273
+ "enable_npu_model_patch": true,
274
+ "load_args": false,
275
+ "load_data_args": false,
276
+ "packing": false,
277
+ "packing_length": null,
278
+ "packing_num_proc": 1,
279
+ "lazy_tokenize": false,
280
+ "use_hf": false,
281
+ "ignore_args_error": false,
282
+ "use_swift_lora": false,
283
+ "freeze_parameters": [],
284
+ "freeze_parameters_regex": null,
285
+ "freeze_parameters_ratio": 0.0,
286
+ "trainable_parameters": [],
287
+ "trainable_parameters_regex": null,
288
+ "freeze_llm": false,
289
+ "freeze_vit": true,
290
+ "freeze_aligner": true,
291
+ "target_modules": [
292
+ "all-linear"
293
+ ],
294
+ "target_regex": null,
295
+ "target_parameters": null,
296
+ "modules_to_save": [],
297
+ "lora_rank": 8,
298
+ "lora_alpha": 32,
299
+ "lora_dropout": 0.05,
300
+ "lora_bias": "none",
301
+ "lora_dtype": null,
302
+ "lorap_lr_ratio": null,
303
+ "use_rslora": false,
304
+ "use_dora": false,
305
+ "lora_ga_batch_size": 2,
306
+ "lora_ga_iters": 2,
307
+ "lora_ga_max_length": 1024,
308
+ "lora_ga_direction": "ArB2r",
309
+ "lora_ga_scale": "stable",
310
+ "lora_ga_stable_gamma": 16,
311
+ "init_weights": true,
312
+ "fourier_n_frequency": 2000,
313
+ "fourier_scaling": 300.0,
314
+ "boft_block_size": 4,
315
+ "boft_block_num": 0,
316
+ "boft_n_butterfly_factor": 1,
317
+ "boft_dropout": 0.0,
318
+ "vera_rank": 256,
319
+ "vera_projection_prng_key": 0,
320
+ "vera_dropout": 0.0,
321
+ "vera_d_initial": 0.1,
322
+ "adapter_act": "gelu",
323
+ "adapter_length": 128,
324
+ "adalora_target_r": 8,
325
+ "adalora_init_r": 12,
326
+ "adalora_tinit": 0,
327
+ "adalora_tfinal": 0,
328
+ "adalora_deltaT": 1,
329
+ "adalora_beta1": 0.85,
330
+ "adalora_beta2": 0.85,
331
+ "adalora_orth_reg_weight": 0.5,
332
+ "llamapro_num_new_blocks": 4,
333
+ "llamapro_num_groups": null,
334
+ "reft_layer_key": null,
335
+ "reft_layers": null,
336
+ "reft_rank": 4,
337
+ "reft_intervention_type": "LoreftIntervention",
338
+ "reft_args": null,
339
+ "swanlab_token": null,
340
+ "swanlab_project": "ms-swift",
341
+ "swanlab_workspace": null,
342
+ "swanlab_exp_name": null,
343
+ "swanlab_notification_method": null,
344
+ "swanlab_webhook_url": null,
345
+ "swanlab_secret": null,
346
+ "swanlab_sender_email": null,
347
+ "swanlab_receiver_email": null,
348
+ "swanlab_smtp_server": null,
349
+ "swanlab_smtp_port": null,
350
+ "swanlab_email_language": "zh",
351
+ "swanlab_mode": "cloud",
352
+ "add_version": false,
353
+ "create_checkpoint_symlink": false,
354
+ "zero_hpz_partition_size": null,
355
+ "deepspeed_autotp_size": null,
356
+ "swift_version": "4.3.0",
357
+ "ckpt_dir": null,
358
+ "rank": 0,
359
+ "global_world_size": 2,
360
+ "local_world_size": 2,
361
+ "model_suffix": "MiniCPM5-1B",
362
+ "model_info": "ModelInfo(model_type='llama', model_dir='/data/leizikun/msp/model/MiniCPM5-1B', torch_dtype=torch.bfloat16, max_model_len=131072, quant_method=None, quant_bits=None, rope_scaling={'rope_theta': 5000000, 'rope_type': 'default'}, is_moe_model=False, is_multimodal=False, config=None, task_type='causal_lm', num_labels=None)",
363
+ "model_meta": "ModelMeta(model_type='llama', model_groups=[ModelGroup(models=[Model(ms_model_id='modelscope/Llama-2-7b-ms', hf_model_id='meta-llama/Llama-2-7b-hf', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='modelscope/Llama-2-13b-ms', hf_model_id='meta-llama/Llama-2-13b-hf', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='modelscope/Llama-2-70b-ms', hf_model_id='meta-llama/Llama-2-70b-hf', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='modelscope/Llama-2-7b-chat-ms', hf_model_id='meta-llama/Llama-2-7b-chat-hf', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='modelscope/Llama-2-13b-chat-ms', hf_model_id='meta-llama/Llama-2-13b-chat-hf', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='modelscope/Llama-2-70b-chat-ms', hf_model_id='meta-llama/Llama-2-70b-chat-hf', model_path=None, ms_revision=None, hf_revision=None)], template='llama', ignore_patterns=['.+\\\\.bin$'], requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='AI-ModelScope/chinese-llama-2-1.3b', hf_model_id='hfl/chinese-llama-2-1.3b', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-llama-2-7b', hf_model_id='hfl/chinese-llama-2-7b', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-llama-2-7b-16k', hf_model_id='hfl/chinese-llama-2-7b-16k', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-llama-2-7b-64k', hf_model_id='hfl/chinese-llama-2-7b-64k', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-llama-2-13b', hf_model_id='hfl/chinese-llama-2-13b', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-llama-2-13b-16k', hf_model_id='hfl/chinese-llama-2-13b-16k', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-alpaca-2-1.3b', hf_model_id='hfl/chinese-alpaca-2-1.3b', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-alpaca-2-7b', hf_model_id='hfl/chinese-alpaca-2-7b', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-alpaca-2-7b-16k', hf_model_id='hfl/chinese-alpaca-2-7b-16k', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-alpaca-2-7b-64k', hf_model_id='hfl/chinese-alpaca-2-7b-64k', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-alpaca-2-13b', hf_model_id='hfl/chinese-alpaca-2-13b', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='AI-ModelScope/chinese-alpaca-2-13b-16k', hf_model_id='hfl/chinese-alpaca-2-13b-16k', model_path=None, ms_revision=None, hf_revision=None)], template='llama', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='AI-ModelScope/Llama-2-7b-AQLM-2Bit-1x16-hf', hf_model_id='ISTA-DASLab/Llama-2-7b-AQLM-2Bit-1x16-hf', model_path=None, ms_revision=None, hf_revision=None)], template='llama', ignore_patterns=None, requires=['transformers>=4.38', 'aqlm', 'torch>=2.2.0'], tags=[]), ModelGroup(models=[Model(ms_model_id='FlagAlpha/Atom-7B', hf_model_id='FlagAlpha/Atom-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='FlagAlpha/Atom-7B-Chat', hf_model_id='FlagAlpha/Atom-7B-Chat', model_path=None, ms_revision=None, hf_revision=None)], template='atom', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='langboat/Mengzi3-13B-Base', hf_model_id='Langboat/Mengzi3-13B-Base', model_path=None, ms_revision=None, hf_revision=None)], template='mengzi', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='AI-ModelScope/NuminaMath-7B-TIR', hf_model_id='AI-MO/NuminaMath-7B-TIR', model_path=None, ms_revision=None, hf_revision=None)], template='numina', ignore_patterns=None, requires=None, tags=['math']), ModelGroup(models=[Model(ms_model_id='Fengshenbang/Ziya2-13B-Base', hf_model_id='IDEA-CCNL/Ziya2-13B-Base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Fengshenbang/Ziya2-13B-Chat', hf_model_id='IDEA-CCNL/Ziya2-13B-Chat', model_path=None, ms_revision=None, hf_revision=None)], template='ziya', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='InfiniAI/Megrez-3b-Instruct', hf_model_id='Infinigence/Megrez-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None)], template='megrez', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='deepseek-ai/deepseek-llm-7b-base', hf_model_id='deepseek-ai/deepseek-llm-7b-base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-llm-7b-chat', hf_model_id='deepseek-ai/deepseek-llm-7b-chat', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-llm-67b-base', hf_model_id='deepseek-ai/deepseek-llm-67b-base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-llm-67b-chat', hf_model_id='deepseek-ai/deepseek-llm-67b-chat', model_path=None, ms_revision=None, hf_revision=None)], template='deepseek', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='deepseek-ai/deepseek-math-7b-base', hf_model_id='deepseek-ai/deepseek-math-7b-base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-math-7b-instruct', hf_model_id='deepseek-ai/deepseek-math-7b-instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-math-7b-rl', hf_model_id='deepseek-ai/deepseek-math-7b-rl', model_path=None, ms_revision=None, hf_revision=None)], template='deepseek', ignore_patterns=None, requires=None, tags=['math']), ModelGroup(models=[Model(ms_model_id='deepseek-ai/deepseek-coder-1.3b-base', hf_model_id='deepseek-ai/deepseek-coder-1.3b-base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-coder-1.3b-instruct', hf_model_id='deepseek-ai/deepseek-coder-1.3b-instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-coder-6.7b-base', hf_model_id='deepseek-ai/deepseek-coder-6.7b-base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-coder-6.7b-instruct', hf_model_id='deepseek-ai/deepseek-coder-6.7b-instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-coder-33b-base', hf_model_id='deepseek-ai/deepseek-coder-33b-base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/deepseek-coder-33b-instruct', hf_model_id='deepseek-ai/deepseek-coder-33b-instruct', model_path=None, ms_revision=None, hf_revision=None)], template='deepseek', ignore_patterns=None, requires=None, tags=['coding']), ModelGroup(models=[Model(ms_model_id='gongjy/MiniMind2', hf_model_id='jingyaogong/MiniMind2', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id=None, hf_model_id='jingyaogong/MiniMind2-Small', model_path=None, ms_revision=None, hf_revision=None)], template='minimind', ignore_patterns=None, requires=['transformers>=4.57.1'], tags=[]), ModelGroup(models=[Model(ms_model_id='LLM-Research/Meta-Llama-3-8B-Instruct', hf_model_id='meta-llama/Meta-Llama-3-8B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3-70B-Instruct', hf_model_id='meta-llama/Meta-Llama-3-70B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3-8B', hf_model_id='meta-llama/Meta-Llama-3-8B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3-70B', hf_model_id='meta-llama/Meta-Llama-3-70B', model_path=None, ms_revision=None, hf_revision=None)], template='llama3', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='swift/Meta-Llama-3-8B-Instruct-GPTQ-Int4', hf_model_id='study-hjt/Meta-Llama-3-8B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='swift/Meta-Llama-3-8B-Instruct-GPTQ-Int8', hf_model_id='study-hjt/Meta-Llama-3-8B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='swift/Meta-Llama-3-8B-Instruct-AWQ', hf_model_id='study-hjt/Meta-Llama-3-8B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='swift/Meta-Llama-3-70B-Instruct-GPTQ-Int4', hf_model_id='study-hjt/Meta-Llama-3-70B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='swift/Meta-Llama-3-70B-Instruct-GPTQ-Int8', hf_model_id='study-hjt/Meta-Llama-3-70B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='swift/Meta-Llama-3-70B-Instruct-AWQ', hf_model_id='study-hjt/Meta-Llama-3-70B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None)], template='llama3', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='ChineseAlpacaGroup/llama-3-chinese-8b-instruct', hf_model_id='hfl/llama-3-chinese-8b-instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='ChineseAlpacaGroup/llama-3-chinese-8b', hf_model_id='hfl/llama-3-chinese-8b', model_path=None, ms_revision=None, hf_revision=None)], template='llama3', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='LLM-Research/Meta-Llama-3.1-8B-Instruct', hf_model_id='meta-llama/Meta-Llama-3.1-8B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-70B-Instruct', hf_model_id='meta-llama/Meta-Llama-3.1-70B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-405B-Instruct', hf_model_id='meta-llama/Meta-Llama-3.1-405B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-8B', hf_model_id='meta-llama/Meta-Llama-3.1-8B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-70B', hf_model_id='meta-llama/Meta-Llama-3.1-70B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-405B', hf_model_id='meta-llama/Meta-Llama-3.1-405B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-70B-Instruct-FP8', hf_model_id='meta-llama/Meta-Llama-3.1-70B-Instruct-FP8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-405B-Instruct-FP8', hf_model_id='meta-llama/Meta-Llama-3.1-405B-Instruct-FP8', model_path=None, ms_revision=None, hf_revision=None)], template='llama3_2', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='LLM-Research/Meta-Llama-3.1-8B-Instruct-BNB-NF4', hf_model_id='hugging-quants/Meta-Llama-3.1-8B-Instruct-BNB-NF4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-70B-Instruct-bnb-4bit', hf_model_id='unsloth/Meta-Llama-3.1-70B-Instruct-bnb-4bit', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-405B-Instruct-BNB-NF4', hf_model_id='hugging-quants/Meta-Llama-3.1-405B-Instruct-BNB-NF4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-8B-Instruct-GPTQ-INT4', hf_model_id='hugging-quants/Meta-Llama-3.1-8B-Instruct-GPTQ-INT4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-70B-Instruct-GPTQ-INT4', hf_model_id='hugging-quants/Meta-Llama-3.1-70B-Instruct-GPTQ-INT4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-405B-Instruct-GPTQ-INT4', hf_model_id='hugging-quants/Meta-Llama-3.1-405B-Instruct-GPTQ-INT4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-8B-Instruct-AWQ-INT4', hf_model_id='hugging-quants/Meta-Llama-3.1-8B-Instruct-AWQ-INT4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-70B-Instruct-AWQ-INT4', hf_model_id='hugging-quants/Meta-Llama-3.1-70B-Instruct-AWQ-INT4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Meta-Llama-3.1-405B-Instruct-AWQ-INT4', hf_model_id='hugging-quants/Meta-Llama-3.1-405B-Instruct-AWQ-INT4', model_path=None, ms_revision=None, hf_revision=None)], template='llama3_2', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='AI-ModelScope/Llama-3.1-Nemotron-70B-Instruct-HF', hf_model_id='nvidia/Llama-3.1-Nemotron-70B-Instruct-HF', model_path=None, ms_revision=None, hf_revision=None)], template='llama3_2', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='AI-ModelScope/Skywork-o1-Open-Llama-3.1-8B', hf_model_id='Skywork/Skywork-o1-Open-Llama-3.1-8B', model_path=None, ms_revision=None, hf_revision=None)], template='skywork_o1', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='LLM-Research/Llama-3.2-1B', hf_model_id='meta-llama/Llama-3.2-1B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Llama-3.2-3B', hf_model_id='meta-llama/Llama-3.2-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Llama-3.2-1B-Instruct', hf_model_id='meta-llama/Llama-3.2-1B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='LLM-Research/Llama-3.2-3B-Instruct', hf_model_id='meta-llama/Llama-3.2-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None)], template='llama3_2', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='LLM-Research/Llama-3.3-70B-Instruct', hf_model_id='meta-llama/Llama-3.3-70B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='unsloth/Llama-3.3-70B-Instruct-bnb-4bit', hf_model_id='unsloth/Llama-3.3-70B-Instruct-bnb-4bit', model_path=None, ms_revision=None, hf_revision=None)], template='llama3_2', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='ZhipuAI/LongWriter-llama3.1-8b', hf_model_id='zai-org/LongWriter-llama3.1-8b', model_path=None, ms_revision=None, hf_revision=None)], template='longwriter_llama', ignore_patterns=None, requires=['transformers>=4.43'], tags=[]), ModelGroup(models=[Model(ms_model_id='deepseek-ai/DeepSeek-R1-Distill-Llama-8B', hf_model_id='deepseek-ai/DeepSeek-R1-Distill-Llama-8B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='deepseek-ai/DeepSeek-R1-Distill-Llama-70B', hf_model_id='deepseek-ai/DeepSeek-R1-Distill-Llama-70B', model_path=None, ms_revision=None, hf_revision=None)], template='deepseek_r1', ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='OpenBMB/MiniCPM5-1B', hf_model_id='openbmb/MiniCPM5-1B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='OpenBMB/MiniCPM5-1B-Base', hf_model_id='openbmb/MiniCPM5-1B-Base', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='OpenBMB/MiniCPM5-1B-SFT', hf_model_id='openbmb/MiniCPM5-1B-SFT', model_path=None, ms_revision=None, hf_revision=None)], template='minicpm5', ignore_patterns=None, requires=['transformers>=5.6'], tags=[]), ModelGroup(models=[Model(ms_model_id='LLM-Research/Reflection-Llama-3.1-70B', hf_model_id='mattshumer/Reflection-Llama-3.1-70B', model_path=None, ms_revision=None, hf_revision=None)], template='reflection', ignore_patterns=None, requires=['transformers>=4.43'], tags=[])], loader=<class 'swift.model.models.llama.LlamaLoader'>, template='minicpm5', model_arch=ModelKeys(arch_name='llama', embedding='model.embed_tokens', module_list='model.layers', lm_head='lm_head', q_proj='model.layers.{}.self_attn.q_proj', k_proj='model.layers.{}.self_attn.k_proj', v_proj='model.layers.{}.self_attn.v_proj', o_proj='model.layers.{}.self_attn.o_proj', attention='model.layers.{}.self_attn', mlp='model.layers.{}.mlp', down_proj='model.layers.{}.mlp.down_proj', qkv_proj=None, qk_proj=None, qa_proj=None, qb_proj=None, kv_proj=None, kva_proj=None, kvb_proj=None), mcore_model_type=None, architectures=['LlamaForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=False, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=5.6'], tags=[])",
364
+ "model_dir": "/data/leizikun/msp/model/MiniCPM5-1B",
365
+ "template_meta": "ChatmlTemplateMeta(template_type='minicpm5', prefix=[], prompt=['<|im_start|>user\\n{{QUERY}}<|im_end|>\\n<|im_start|>assistant\\n'], chat_sep=['<|im_end|>\\n'], suffix=['<|im_end|>\\n'], template_cls=<class 'swift.template.base.Template'>, system_prefix=['<|im_start|>system\\n{{SYSTEM}}<|im_end|>\\n'], default_system=None, auto_add_bos=True, stop_words=[], agent_template='minicpm5', is_thinking=True, thinking_prefix='<think>\\n', non_thinking_prefix='<think>\\n\\n</think>\\n\\n', history_thinking_prefix='')",
366
+ "_val_dataset_exists": false,
367
+ "hub": "<class 'swift.hub.hub.MSHub'>",
368
+ "evaluation_strategy": "epoch",
369
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/leizikun/msp/script/occupy/output/minicpm5-1b-fullft-neko', per_device_train_batch_size=4, num_train_epochs=5.0, max_steps=-1, learning_rate=2e-05, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_steps=0.03, optim=<OptimizerNames.ADAMW_TORCH_FUSED: 'adamw_torch_fused'>, optim_args=None, weight_decay=0.01, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, optim_target_modules=None, gradient_accumulation_steps=4, average_tokens_across_devices=None, max_grad_norm=1.0, label_smoothing_factor=0.0, bf16=True, fp16=False, bf16_full_eval=False, fp16_full_eval=False, tf32=None, gradient_checkpointing=True, gradient_checkpointing_kwargs=None, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, use_liger_kernel=False, liger_kernel_config=None, use_cache=False, neftune_noise_alpha=None, torch_empty_cache_steps=None, auto_find_batch_size=False, logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_steps=10, logging_first_step=True, log_on_each_node=True, logging_nan_inf_filter=True, include_num_input_tokens_seen=None, log_level='passive', log_level_replica='warning', disable_tqdm=False, report_to=['wandb'], run_name='minicpm5-1b-fullft-2a800-z2-bs4-ga4-4096', project='huggingface', trackio_space_id=None, trackio_bucket_id=None, trackio_static_space_id=None, eval_strategy=<IntervalStrategy.NO: 'no'>, eval_steps=None, eval_delay=0, per_device_eval_batch_size=4, prediction_loss_only=False, eval_on_start=False, eval_do_concat_batches=True, eval_use_gather_object=False, eval_accumulation_steps=None, include_for_metrics=[], batch_eval_metrics=False, save_only_model=True, save_strategy=<SaveStrategy.EPOCH: 'epoch'>, save_steps=500, save_on_each_node=False, save_total_limit=3, enable_jit_checkpoint=False, push_to_hub=False, hub_token=None, hub_private_repo=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_always_push=False, hub_revision=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, restore_callback_states_from_checkpoint=False, full_determinism=False, seed=42, data_seed=42, use_cpu=False, accelerator_config=AcceleratorConfig(split_batches=False, dispatch_batches=False, even_batches=True, use_seedable_sampler=True, non_blocking=False, gradient_accumulation_kwargs=None, use_configured_state=False), parallelism_config=None, dataloader_drop_last=False, dataloader_num_workers=4, dataloader_pin_memory=True, dataloader_persistent_workers=False, dataloader_prefetch_factor=2, remove_unused_columns=False, label_names=None, train_sampling_strategy='random', length_column_name='length', ddp_find_unused_parameters=False, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, ddp_static_graph=None, ddp_backend='nccl', ddp_timeout=18000000, fsdp=[], fsdp_config=None, deepspeed={'bf16': {'enabled': 'auto'}, 'zero_optimization': {'stage': 2, 'offload_optimizer': {'device': 'none', 'pin_memory': True}, 'allgather_partitions': True, 'allgather_bucket_size': 500000000.0, 'overlap_comm': True, 'reduce_scatter': True, 'reduce_bucket_size': 500000000.0, 'contiguous_gradients': True}, 'gradient_accumulation_steps': 'auto', 'gradient_clipping': 'auto', 'train_batch_size': 'auto', 'train_micro_batch_size_per_gpu': 'auto', 'wall_clock_breakdown': False}, debug=[], skip_memory_metrics=True, do_train=False, do_eval=False, do_predict=False, resume_from_checkpoint=None, warmup_ratio=0.03, logging_dir='/data/leizikun/msp/script/occupy/output/minicpm5-1b-fullft-neko/runs', local_rank=0, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, tuner_backend='peft', vit_gradient_checkpointing=False, router_aux_loss_coef=0.0, enable_dft_loss=False, enable_channel_loss=False, safe_serialization=True, max_shard_size='5GB', check_model=True, acc_strategy='token', train_dataloader_shuffle=True, group_by_length=False, max_epochs=None, aligner_lr=None, vit_lr=None, use_logits_to_keep=None, ds3_gather_for_generation=True, resume_only_model=False, optimizer=None, loss_type=None, eval_metric=None, callbacks=[], early_stop_interval=None, eval_use_evalscope=False, eval_dataset=[], eval_dataset_args=None, eval_limit=None, eval_generation_config=None, extra_eval_args=None, tuner_type='full', use_galore=False, galore_target_modules=None, galore_rank=128, galore_update_proj_gap=50, galore_scale=1.0, galore_proj_type='std', galore_optim_per_parameter=False, galore_with_embedding=False, galore_quantization=False, galore_proj_quant=False, galore_proj_bits=4, galore_proj_group_size=256, galore_cos_threshold=0.4, galore_gamma_proj=2, galore_queue_size=5, lisa_activated_layers=0, lisa_step_interval=20, use_flash_ckpt=False)"
370
+ }
chat_template.jinja ADDED
@@ -0,0 +1,179 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{- bos_token }}{%- if tools %}
2
+ {%- set tool_definitions %}
3
+ {{- "# Tools\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
4
+ {%- for tool in tools %}
5
+ {{- "\n" }}
6
+ {{- tool | tojson(ensure_ascii=False) }}
7
+ {%- endfor %}
8
+ {{- '\n</tools>\n\nTool usage guidelines:\n- You may call zero or more functions. If no function calls are needed, just answer normally and do not include any <function ... </function>.\n- When calling a function, return an XML object within <function ... </function> using:\n<function name="function-name"><param name="param-name">param-value</param></function>\n- param-value may be multi-line. If it contains <, & or newline characters, wrap it in a CDATA block: <param name="param-name"><![CDATA[...multi-line value...]]></param>' }}
9
+ {%- endset %}
10
+
11
+ {{- '<|im_start|>system\n' }}
12
+ {%- if messages[0].role == 'system' %}
13
+ {%- if '<tool_def_sep>' in messages[0].content %}
14
+ {{- messages[0].content.replace('<tool_def_sep>', tool_definitions) }}
15
+ {%- else %}
16
+ {{- messages[0].content + '\n\n' + tool_definitions }}
17
+ {%- endif %}
18
+ {%- else %}
19
+ {{- tool_definitions.lstrip() }}
20
+ {%- endif %}
21
+ {{- '<|im_end|>\n' }}
22
+ {%- else %}
23
+ {%- if messages[0].role == 'system' %}
24
+ {{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
25
+ {%- endif %}
26
+ {%- endif %}
27
+ {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
28
+ {%- for message in messages[::-1] %}
29
+ {%- set index = (messages|length - 1) - loop.index0 %}
30
+ {%- if ns.multi_step_tool and message.role == "user" and message.content is string and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}
31
+ {%- set ns.multi_step_tool = false %}
32
+ {%- set ns.last_query_index = index %}
33
+ {%- endif %}
34
+ {%- endfor %}
35
+ {%- for message in messages %}
36
+ {%- if message.content is string %}
37
+ {%- set content = message.content %}
38
+ {%- else %}
39
+ {%- set content = '' %}
40
+ {%- endif %}
41
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
42
+ {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
43
+ {%- elif message.role == "assistant" %}
44
+ {%- set reasoning_content = '' %}
45
+ {%- if message.reasoning_content is string %}
46
+ {%- set reasoning_content = message.reasoning_content %}
47
+ {%- else %}
48
+ {%- if '</think>' in content %}
49
+ {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
50
+ {%- set content = content.split('</think>')[-1].lstrip('\n') %}
51
+ {%- endif %}
52
+ {%- endif %}
53
+
54
+ {%- if message.tool_calls %}
55
+ {%- set content_parts = content.split('<tool_sep>') %}
56
+ {%- set processed_content = content_parts[0] %}
57
+ {%- set tool_calls_count = message.tool_calls|length %}
58
+ {%- set tool_sep_count = content_parts|length - 1 %}
59
+ {%- set min_count = [tool_calls_count, tool_sep_count]|min %}
60
+
61
+ {%- for i in range(1, content_parts|length) %}
62
+ {%- set tool_index = i - 1 %}
63
+ {%- if tool_index < tool_calls_count %}
64
+ {%- set tool_call = message.tool_calls[tool_index] %}
65
+ {%- if tool_call.function %}
66
+ {%- set tool_call = tool_call.function %}
67
+ {%- endif %}
68
+ {%- set single_tool_xml %}
69
+ {{- '<function name="' ~ tool_call.name ~ '">' }}
70
+ {%- if tool_call.arguments %}
71
+ {%- set args_dict = tool_call.arguments %}
72
+ {%- for param_name, param_value in args_dict.items() %}
73
+ {{- '<param name="' ~ param_name ~ '">' }}
74
+ {%- if param_value is string and ('<' in param_value or '&' in param_value or '\n' in param_value) %}
75
+ {{- '<![CDATA[' + param_value + ']]>' }}
76
+ {%- else %}
77
+ {{- param_value }}
78
+ {%- endif %}
79
+ {{- '</param>' }}
80
+ {%- endfor %}
81
+ {%- endif %}
82
+ {{- '</function>' }}
83
+ {%- endset %}
84
+ {%- set processed_content = processed_content + single_tool_xml + content_parts[i] %}
85
+ {%- else %}
86
+ {%- set processed_content = processed_content + content_parts[i] %}
87
+ {%- endif %}
88
+ {%- endfor %}
89
+
90
+ {%- if tool_calls_count > tool_sep_count %}
91
+ {%- for remaining_index in range(tool_sep_count, tool_calls_count) %}
92
+ {%- set tool_call = message.tool_calls[remaining_index] %}
93
+ {%- if tool_call.function %}
94
+ {%- set tool_call = tool_call.function %}
95
+ {%- endif %}
96
+ {%- set remaining_tool_xml %}
97
+ {{- '<function name="' ~ tool_call.name ~ '">' }}
98
+ {%- if tool_call.arguments %}
99
+ {%- set args_dict = tool_call.arguments %}
100
+ {%- for param_name, param_value in args_dict.items() %}
101
+ {{- '<param name="' ~ param_name ~ '">' }}
102
+ {%- if param_value is string and ('<' in param_value or '&' in param_value or '\n' in param_value) %}
103
+ {{- '<![CDATA[' + param_value + ']]>' }}
104
+ {%- else %}
105
+ {{- param_value }}
106
+ {%- endif %}
107
+ {{- '</param>' }}
108
+ {%- endfor %}
109
+ {%- endif %}
110
+ {{- '</function>' }}
111
+ {%- endset %}
112
+ {%- set processed_content = processed_content + remaining_tool_xml %}
113
+ {%- endfor %}
114
+ {%- endif %}
115
+
116
+ {%- set content = processed_content %}
117
+ {%- endif %}
118
+
119
+ {%- if loop.index0 > ns.last_query_index %}
120
+ {%- if reasoning_content %}
121
+ {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
122
+ {%- else %}
123
+ {{- '<|im_start|>' + message.role + '\n' + content }}
124
+ {%- endif %}
125
+ {%- else %}
126
+ {{- '<|im_start|>' + message.role + '\n' + content }}
127
+ {%- endif %}
128
+
129
+ {%- if message.tool_calls and not has_tool_sep %}
130
+ {%- for tool_call in message.tool_calls %}
131
+ {%- if (loop.first and content) or (not loop.first) %}
132
+ {{- '\n' }}
133
+ {%- endif %}
134
+ {%- if tool_call.function %}
135
+ {%- set tool_call = tool_call.function %}
136
+ {%- endif %}
137
+ {{- '<function name="' ~ tool_call.name ~ '">' }}
138
+ {%- if tool_call.arguments %}
139
+ {%- set args_dict = tool_call.arguments %}
140
+ {%- for param_name, param_value in args_dict.items() %}
141
+ {{- '<param name="' ~ param_name ~ '">' }}
142
+ {%- if param_value is string and ('<' in param_value or '&' in param_value or '\n' in param_value) %}
143
+ {{- '<![CDATA[' + param_value + ']]>' }}
144
+ {%- else %}
145
+ {{- param_value }}
146
+ {%- endif %}
147
+ {{- '</param>' }}
148
+ {%- endfor %}
149
+ {%- endif %}
150
+ {{- '</function>' }}
151
+ {%- endfor %}
152
+ {%- endif %}
153
+ {{- '<|im_end|>\n' }}
154
+ {%- elif message.role == "tool" %}
155
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
156
+ {{- '<|im_start|>user' }}
157
+ {%- endif %}
158
+ {{- '\n<tool_response>\n' }}
159
+ {%- if message.content is string %}
160
+ {{- content }}
161
+ {%- else %}
162
+ {{- message.content | tojson(ensure_ascii=False) }}
163
+ {%- endif %}
164
+ {{- '\n</tool_response>' }}
165
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
166
+ {{- '<|im_end|>\n' }}
167
+ {%- endif %}
168
+ {%- endif %}
169
+ {%- endfor %}
170
+ {%- if add_generation_prompt %}
171
+ {{- '<|im_start|>assistant\n' }}
172
+ {%- if enable_thinking is defined %}
173
+ {%- if enable_thinking is false %}
174
+ {{- '<think>\n\n</think>\n\n' }}
175
+ {%- elif enable_thinking is true %}
176
+ {{- '<think>\n' }}
177
+ {%- endif %}
178
+ {%- endif %}
179
+ {%- endif %}
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "dtype": "bfloat16",
9
+ "eos_token_id": 1,
10
+ "head_dim": 128,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 1536,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 4608,
15
+ "max_position_embeddings": 131072,
16
+ "mlp_bias": false,
17
+ "model_type": "llama",
18
+ "num_attention_heads": 16,
19
+ "num_hidden_layers": 24,
20
+ "num_key_value_heads": 2,
21
+ "pad_token_id": 1,
22
+ "pretraining_tp": 1,
23
+ "rms_norm_eps": 1e-06,
24
+ "rope_parameters": {
25
+ "rope_theta": 5000000,
26
+ "rope_type": "default"
27
+ },
28
+ "tie_word_embeddings": false,
29
+ "transformers_version": "5.10.2",
30
+ "use_cache": false,
31
+ "vocab_size": 130560
32
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 130073
8
+ ],
9
+ "pad_token_id": 1,
10
+ "temperature": 0.9,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.10.2"
13
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56e5c04e2ade5b2fc32fdce0ea907703b81c49154b9c97dc71dee0800203c018
3
+ size 2161290944
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": null,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<s>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "</s>",
7
+ "is_local": true,
8
+ "legacy": true,
9
+ "local_files_only": false,
10
+ "model_max_length": 1000000000000000019884624838656,
11
+ "pad_token": "</s>",
12
+ "sp_model_kwargs": {},
13
+ "spaces_between_special_tokens": false,
14
+ "tokenizer_class": "PreTrainedTokenizerFast",
15
+ "unk_token": "<unk>",
16
+ "use_default_system_prompt": false
17
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0e4273416626fb87df306612ea41f4481d5c94232c2771f7b1a1fcee32338a6
3
+ size 8593