Ibisbill
/

stage3_SimpleRL_lr_1e5_epoch2

Text Generation

text-generation-inference

Model card Files Files and versions

Ibisbill commited on May 28, 2025

Commit

3459968

·

verified ·

1 Parent(s): ef789bb

Upload config

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "/map-vepfs/ziyu/reasoning_tr/saves/stage3_7b_baseline/Qwen-2.5-Math-7B-SimpleRL-Zoo_2epoch/stage3_combined_50k",
   "architectures": [
-    "Qwen2Model"
   ],
   "attention_dropout": 0.0,
   "eos_token_id": 151643,
@@ -21,7 +21,7 @@
   "rope_theta": 10000,
   "sliding_window": null,
   "tie_word_embeddings": false,
-  "torch_dtype": "float32",
   "transformers_version": "4.49.0",
   "use_cache": false,
   "use_mrope": false,

 {
   "_name_or_path": "/map-vepfs/ziyu/reasoning_tr/saves/stage3_7b_baseline/Qwen-2.5-Math-7B-SimpleRL-Zoo_2epoch/stage3_combined_50k",
   "architectures": [
+    "Qwen2ForCausalLM"
   ],
   "attention_dropout": 0.0,
   "eos_token_id": 151643,
   "rope_theta": 10000,
   "sliding_window": null,
   "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.49.0",
   "use_cache": false,
   "use_mrope": false,