Ba2han commited on
Commit
899104a
·
verified ·
1 Parent(s): b137307

Training in progress, step 500

Browse files
Files changed (5) hide show
  1. README.md +4 -4
  2. chat_template.jinja +3 -3
  3. config.json +8 -16
  4. model.safetensors +2 -2
  5. training_args.bin +1 -1
README.md CHANGED
@@ -1,18 +1,18 @@
1
  ---
2
- base_model: unsloth/Qwen3.5-4B-Base
3
  library_name: transformers
4
  model_name: augment-multi-qwen
5
  tags:
6
  - generated_from_trainer
 
7
  - unsloth
8
  - sft
9
- - trl
10
  licence: license
11
  ---
12
 
13
  # Model Card for augment-multi-qwen
14
 
15
- This model is a fine-tuned version of [unsloth/Qwen3.5-4B-Base](https://huggingface.co/unsloth/Qwen3.5-4B-Base).
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
18
  ## Quick start
@@ -28,7 +28,7 @@ print(output["generated_text"])
28
 
29
  ## Training procedure
30
 
31
- [<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/batuhan409/huggingface/runs/d2zulkmg)
32
 
33
 
34
  This model was trained with SFT.
 
1
  ---
2
+ base_model: unsloth/Qwen3.5-2B-Base
3
  library_name: transformers
4
  model_name: augment-multi-qwen
5
  tags:
6
  - generated_from_trainer
7
+ - trl
8
  - unsloth
9
  - sft
 
10
  licence: license
11
  ---
12
 
13
  # Model Card for augment-multi-qwen
14
 
15
+ This model is a fine-tuned version of [unsloth/Qwen3.5-2B-Base](https://huggingface.co/unsloth/Qwen3.5-2B-Base).
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
18
  ## Quick start
 
28
 
29
  ## Training procedure
30
 
31
+ [<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/batuhan409/huggingface/runs/61dp66cf)
32
 
33
 
34
  This model was trained with SFT.
chat_template.jinja CHANGED
@@ -149,10 +149,10 @@
149
  {%- endfor %}
150
  {%- if add_generation_prompt %}
151
  {{- '<|im_start|>assistant\n' }}
152
- {%- if enable_thinking is defined and enable_thinking is false %}
153
- {{- '<think>\n\n</think>\n\n' }}
154
- {%- else %}
155
  {{- '<think>\n' }}
 
 
156
  {%- endif %}
157
  {%- endif %}
158
  {#- Unsloth fixes - developer role, tool calling #}
 
149
  {%- endfor %}
150
  {%- if add_generation_prompt %}
151
  {{- '<|im_start|>assistant\n' }}
152
+ {%- if enable_thinking is defined and enable_thinking is true %}
 
 
153
  {{- '<think>\n' }}
154
+ {%- else %}
155
+ {{- '<think>\n\n</think>\n\n' }}
156
  {%- endif %}
157
  {%- endif %}
158
  {#- Unsloth fixes - developer role, tool calling #}
config.json CHANGED
@@ -5,7 +5,7 @@
5
  "dtype": "bfloat16",
6
  "eos_token_id": 248044,
7
  "image_token_id": 248056,
8
- "model_name": "unsloth/Qwen3.5-4B-Base",
9
  "model_type": "qwen3_5",
10
  "pad_token_id": 248055,
11
  "text_config": {
@@ -18,9 +18,9 @@
18
  "full_attention_interval": 4,
19
  "head_dim": 256,
20
  "hidden_act": "silu",
21
- "hidden_size": 2560,
22
  "initializer_range": 0.02,
23
- "intermediate_size": 9216,
24
  "layer_types": [
25
  "linear_attention",
26
  "linear_attention",
@@ -45,20 +45,12 @@
45
  "linear_attention",
46
  "linear_attention",
47
  "linear_attention",
48
- "full_attention",
49
- "linear_attention",
50
- "linear_attention",
51
- "linear_attention",
52
- "full_attention",
53
- "linear_attention",
54
- "linear_attention",
55
- "linear_attention",
56
  "full_attention"
57
  ],
58
  "linear_conv_kernel_dim": 4,
59
  "linear_key_head_dim": 128,
60
  "linear_num_key_heads": 16,
61
- "linear_num_value_heads": 32,
62
  "linear_value_head_dim": 128,
63
  "mamba_ssm_dtype": "float32",
64
  "max_position_embeddings": 262144,
@@ -66,9 +58,9 @@
66
  "model_type": "qwen3_5_text",
67
  "mtp_num_hidden_layers": 1,
68
  "mtp_use_dedicated_embeddings": false,
69
- "num_attention_heads": 16,
70
- "num_hidden_layers": 32,
71
- "num_key_value_heads": 4,
72
  "pad_token_id": null,
73
  "partial_rotary_factor": 0.25,
74
  "rms_norm_eps": 1e-06,
@@ -105,7 +97,7 @@
105
  "model_type": "qwen3_5",
106
  "num_heads": 16,
107
  "num_position_embeddings": 2304,
108
- "out_hidden_size": 2560,
109
  "patch_size": 16,
110
  "spatial_merge_size": 2,
111
  "temporal_patch_size": 2
 
5
  "dtype": "bfloat16",
6
  "eos_token_id": 248044,
7
  "image_token_id": 248056,
8
+ "model_name": "unsloth/Qwen3.5-2B-Base",
9
  "model_type": "qwen3_5",
10
  "pad_token_id": 248055,
11
  "text_config": {
 
18
  "full_attention_interval": 4,
19
  "head_dim": 256,
20
  "hidden_act": "silu",
21
+ "hidden_size": 2048,
22
  "initializer_range": 0.02,
23
+ "intermediate_size": 6144,
24
  "layer_types": [
25
  "linear_attention",
26
  "linear_attention",
 
45
  "linear_attention",
46
  "linear_attention",
47
  "linear_attention",
 
 
 
 
 
 
 
 
48
  "full_attention"
49
  ],
50
  "linear_conv_kernel_dim": 4,
51
  "linear_key_head_dim": 128,
52
  "linear_num_key_heads": 16,
53
+ "linear_num_value_heads": 16,
54
  "linear_value_head_dim": 128,
55
  "mamba_ssm_dtype": "float32",
56
  "max_position_embeddings": 262144,
 
58
  "model_type": "qwen3_5_text",
59
  "mtp_num_hidden_layers": 1,
60
  "mtp_use_dedicated_embeddings": false,
61
+ "num_attention_heads": 8,
62
+ "num_hidden_layers": 24,
63
+ "num_key_value_heads": 2,
64
  "pad_token_id": null,
65
  "partial_rotary_factor": 0.25,
66
  "rms_norm_eps": 1e-06,
 
97
  "model_type": "qwen3_5",
98
  "num_heads": 16,
99
  "num_position_embeddings": 2304,
100
+ "out_hidden_size": 2048,
101
  "patch_size": 16,
102
  "spatial_merge_size": 2,
103
  "temporal_patch_size": 2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ad37fbb1ddb6fb65e603dbd401eaefe6d8ff342961fe7b4405241ef4283be759
3
- size 9079183800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9d7dc46d599cf286ba68d2e794f8c3e5ffd29ddd33efbe976203e0401d7ac4b
3
+ size 4426983440
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6ae3ccc52bc9f93528e8f46aae3537216dab67a557a313fa59046ba75eeb9185
3
  size 5713
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:203650bea988f3f34ce73d678d87d9717e4df8feef71208ad0d37a3fcb6ae426
3
  size 5713