lukedai commited on
Commit
b352e0a
·
verified ·
1 Parent(s): 51e69ee

Model save

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen2.5-0.5B-Instruct
3
+ library_name: transformers
4
+ model_name: Qwen2.5-1.5B-Open-R1-Distill-sft-debug
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for Qwen2.5-1.5B-Open-R1-Distill-sft-debug
13
+
14
+ This model is a fine-tuned version of [Qwen/Qwen2.5-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="lukedai/Qwen2.5-1.5B-Open-R1-Distill-sft-debug", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/hanbrisk-freelancer/huggingface/runs/m5vr0uy8)
31
+
32
+
33
+ This model was trained with SFT.
34
+
35
+ ### Framework versions
36
+
37
+ - TRL: 0.16.0.dev0
38
+ - Transformers: 4.49.0
39
+ - Pytorch: 2.5.1
40
+ - Datasets: 3.3.2
41
+ - Tokenizers: 0.21.0
42
+
43
+ ## Citations
44
+
45
+
46
+
47
+ Cite TRL as:
48
+
49
+ ```bibtex
50
+ @misc{vonwerra2022trl,
51
+ title = {{TRL: Transformer Reinforcement Learning}},
52
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
53
+ year = 2020,
54
+ journal = {GitHub repository},
55
+ publisher = {GitHub},
56
+ howpublished = {\url{https://github.com/huggingface/trl}}
57
+ }
58
+ ```
adapter_config.json CHANGED
@@ -23,8 +23,8 @@
23
  "rank_pattern": {},
24
  "revision": null,
25
  "target_modules": [
26
- "q_proj",
27
- "v_proj"
28
  ],
29
  "task_type": "CAUSAL_LM",
30
  "use_dora": false,
 
23
  "rank_pattern": {},
24
  "revision": null,
25
  "target_modules": [
26
+ "v_proj",
27
+ "q_proj"
28
  ],
29
  "task_type": "CAUSAL_LM",
30
  "use_dora": false,
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_flos": 1171084304973824.0,
3
+ "train_loss": 0.0,
4
+ "train_runtime": 2.6815,
5
+ "train_samples": 100,
6
+ "train_samples_per_second": 32.444,
7
+ "train_steps_per_second": 3.729
8
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_flos": 1171084304973824.0,
3
+ "train_loss": 0.0,
4
+ "train_runtime": 2.6815,
5
+ "train_samples": 100,
6
+ "train_samples_per_second": 32.444,
7
+ "train_steps_per_second": 3.729
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9195402298850575,
5
+ "eval_steps": 500,
6
+ "global_step": 10,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.45977011494252873,
13
+ "grad_norm": 0.2585427165031433,
14
+ "learning_rate": 3.140708399750594e-05,
15
+ "loss": 1.0544,
16
+ "step": 5
17
+ },
18
+ {
19
+ "epoch": 0.9195402298850575,
20
+ "grad_norm": 0.2384282946586609,
21
+ "learning_rate": 5e-06,
22
+ "loss": 0.9716,
23
+ "step": 10
24
+ },
25
+ {
26
+ "epoch": 0.9195402298850575,
27
+ "step": 10,
28
+ "total_flos": 1171084304973824.0,
29
+ "train_loss": 0.0,
30
+ "train_runtime": 2.6815,
31
+ "train_samples_per_second": 32.444,
32
+ "train_steps_per_second": 3.729
33
+ }
34
+ ],
35
+ "logging_steps": 5,
36
+ "max_steps": 10,
37
+ "num_input_tokens_seen": 0,
38
+ "num_train_epochs": 1,
39
+ "save_steps": 100,
40
+ "stateful_callbacks": {
41
+ "TrainerControl": {
42
+ "args": {
43
+ "should_epoch_stop": false,
44
+ "should_evaluate": false,
45
+ "should_log": false,
46
+ "should_save": true,
47
+ "should_training_stop": true
48
+ },
49
+ "attributes": {}
50
+ }
51
+ },
52
+ "total_flos": 1171084304973824.0,
53
+ "train_batch_size": 1,
54
+ "trial_name": null,
55
+ "trial_params": null
56
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:264120d7cab64bb20fb6ff19360c72809d6dc10ae8d442d23a8e3a268088ba76
3
  size 7544
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ffd9e01accbb5447df72e3174e77c18a2ec9b0cd2185e8a85141980ba5402bd1
3
  size 7544