初始化项目,由ModelHub XC社区提供模型

Model: christinakopi/thinkprm-full-trl
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-13 15:53:16 +08:00
commit d67e730186
39 changed files with 807 additions and 0 deletions

View File

@@ -0,0 +1 @@
{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% set ns = namespace(is_first=false, is_tool=false, is_output_first=true, system_prompt='') %}{%- for message in messages %}{%- if message['role'] == 'system' %}{% set ns.system_prompt = message['content'] %}{%- endif %}{%- endfor %}{{bos_token}}{{ns.system_prompt}}{%- for message in messages %}{%- if message['role'] == 'user' %}{%- set ns.is_tool = false -%}{{'<User>' + message['content']}}{%- endif %}{%- if message['role'] == 'assistant' and message['content'] is none %}{%- set ns.is_tool = false -%}{%- for tool in message['tool_calls']%}{%- if not ns.is_first %}{{'<Assistant><tool▁calls▁begin><tool▁call▁begin>' + tool['type'] + '<tool▁sep>' + tool['function']['name'] + '\n' + '```json' + '\n' + tool['function']['arguments'] + '\n' + '```' + '<tool▁call▁end>'}}{%- set ns.is_first = true -%}{%- else %}{{'\n' + '<tool▁call▁begin>' + tool['type'] + '<tool▁sep>' + tool['function']['name'] + '\n' + '```json' + '\n' + tool['function']['arguments'] + '\n' + '```' + '<tool▁call▁end>'}}{{'<tool▁calls▁end><end▁of▁sentence>'}}{%- endif %}{%- endfor %}{%- endif %}{%- if message['role'] == 'assistant' and message['content'] is not none %}{%- if ns.is_tool %}{{'<tool▁outputs▁end>' + message['content'] + '<end▁of▁sentence>'}}{%- set ns.is_tool = false -%}{%- else %}{% set content = message['content'] %}{% if '</think>' in content %}{% set content = content.split('</think>')[-1] %}{% endif %}{{'<Assistant>' + content + '<end▁of▁sentence>'}}{%- endif %}{%- endif %}{%- if message['role'] == 'tool' %}{%- set ns.is_tool = true -%}{%- if ns.is_output_first %}{{'<tool▁outputs▁begin><tool▁output▁begin>' + message['content'] + '<tool▁output▁end>'}}{%- set ns.is_output_first = false %}{%- else %}{{'\n<tool▁output▁begin>' + message['content'] + '<tool▁output▁end>'}}{%- endif %}{%- endif %}{%- endfor -%}{% if ns.is_tool %}{{'<tool▁outputs▁end>'}}{% endif %}{% if add_generation_prompt and not ns.is_tool %}{{'<Assistant><think>\n'}}{% endif %}

62
checkpoint-64/config.json Normal file
View File

@@ -0,0 +1,62 @@
{
"architectures": [
"Qwen2ForCausalLM"
],
"attention_dropout": 0.0,
"bos_token_id": 151646,
"dtype": "bfloat16",
"eos_token_id": 151643,
"hidden_act": "silu",
"hidden_size": 1536,
"initializer_range": 0.02,
"intermediate_size": 8960,
"layer_types": [
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention",
"full_attention"
],
"max_position_embeddings": 131072,
"max_window_layers": 21,
"model_type": "qwen2",
"num_attention_heads": 12,
"num_hidden_layers": 28,
"num_key_value_heads": 2,
"pad_token_id": 151643,
"rms_norm_eps": 1e-06,
"rope_parameters": {
"rope_theta": 10000,
"rope_type": "default"
},
"sliding_window": null,
"tie_word_embeddings": false,
"transformers_version": "5.5.0",
"use_cache": false,
"use_mrope": false,
"use_sliding_window": false,
"vocab_size": 151936
}

View File

@@ -0,0 +1,12 @@
{
"_from_model_config": true,
"bos_token_id": 151646,
"do_sample": true,
"eos_token_id": [
151643
],
"pad_token_id": 151643,
"temperature": 0.6,
"top_p": 0.95,
"transformers_version": "5.5.0"
}

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:80b03bce94ee75fd40e7e0634cabf03b5ae45b09f4994f9c80d36c32d6c55c5d
size 3554214752

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:9a6a08938e221b78f29b51efd7fcef340d6be254e56950bb4b625c8d34efce6a
size 7108644139

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:1a6ec98147904582f9073debdd2d46db8be81f1fae8b7e30a4fb33c0724bf01a
size 15024

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:53345026ed0535abd8cf6a0e1ecbdbf9a4d5ef86d37a344cbca5ef2567e6cc18
size 15024

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:cb45d6dbfeb6c6a340bea2c45d80cd63df0bd6b093ca44076be3a394a5390059
size 15024

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:184eba8fad81b78f577f59c07102dc1253cd2cdbd814ae3d20c579d2332fc55c
size 15024

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:38aae7f031adb4bca6603aca638a1be57deaca2e29f6d25a07754ffcad18fba3
size 1064

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:322664cdc3082b6eba003af5228a77ca1d7936d402e584ecde8f15d3d98bdb72
size 11421911

View File

@@ -0,0 +1,13 @@
{
"backend": "tokenizers",
"bos_token": "<begin▁of▁sentence>",
"clean_up_tokenization_spaces": false,
"eos_token": "<end▁of▁sentence>",
"is_local": false,
"legacy": true,
"model_max_length": 16384,
"pad_token": "<end▁of▁sentence>",
"sp_model_kwargs": {},
"tokenizer_class": "TokenizersBackend",
"unk_token": null
}

View File

@@ -0,0 +1,94 @@
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 100,
"global_step": 64,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"entropy": 0.4654052224010229,
"epoch": 0.32,
"grad_norm": 0.048828125,
"learning_rate": 6e-05,
"loss": 0.30594801902770996,
"mean_token_accuracy": 0.9038856193423271,
"num_tokens": 460568.0,
"step": 10
},
{
"entropy": 0.42793196588754656,
"epoch": 0.64,
"grad_norm": 0.050537109375,
"learning_rate": 6e-05,
"loss": 0.2569255828857422,
"mean_token_accuracy": 0.9156977899372578,
"num_tokens": 916064.0,
"step": 20
},
{
"entropy": 0.41921926364302636,
"epoch": 0.96,
"grad_norm": 0.043701171875,
"learning_rate": 6e-05,
"loss": 0.26015210151672363,
"mean_token_accuracy": 0.9139690645039081,
"num_tokens": 1373088.0,
"step": 30
},
{
"entropy": 0.38780535757541656,
"epoch": 1.256,
"grad_norm": 0.042724609375,
"learning_rate": 6e-05,
"loss": 0.1978880524635315,
"mean_token_accuracy": 0.9332824486332971,
"num_tokens": 1799690.0,
"step": 40
},
{
"entropy": 0.3551080597564578,
"epoch": 1.576,
"grad_norm": 0.041259765625,
"learning_rate": 6e-05,
"loss": 0.18691558837890626,
"mean_token_accuracy": 0.9357104673981667,
"num_tokens": 2257096.0,
"step": 50
},
{
"entropy": 0.34900726228952406,
"epoch": 1.896,
"grad_norm": 0.042236328125,
"learning_rate": 6e-05,
"loss": 0.1920093059539795,
"mean_token_accuracy": 0.9355965360999108,
"num_tokens": 2710250.0,
"step": 60
}
],
"logging_steps": 10,
"max_steps": 96,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 100,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 2.6497787881324544e+16,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}

View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:8b6ba99fbb102272b4ce278ab05ae0e477eb8ff8bab9f89f281c9f2c3dce6e1d
size 5304