choco800 commited on
Commit
b701cac
·
verified ·
1 Parent(s): 1e7695c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +64 -12
README.md CHANGED
@@ -1,21 +1,73 @@
1
  ---
2
  base_model: Qwen/Qwen3-4B-Instruct-2507
3
- tags:
4
- - text-generation-inference
5
- - transformers
6
- - unsloth
7
- - qwen3
8
- license: apache-2.0
9
  language:
10
  - en
 
 
 
 
 
 
 
 
11
  ---
12
 
13
- # Uploaded finetuned model
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
 
15
- - **Developed by:** choco800
16
- - **License:** apache-2.0
17
- - **Finetuned from model :** Qwen/Qwen3-4B-Instruct-2507
 
 
 
 
 
18
 
19
- This qwen3 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
20
 
21
- [<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
 
 
1
  ---
2
  base_model: Qwen/Qwen3-4B-Instruct-2507
3
+ datasets:
4
+ - u-10bei/dbbench_sft_dataset_react
 
 
 
 
5
  language:
6
  - en
7
+ license: apache-2.0
8
+ pipeline_tag: text-generation
9
+ tags:
10
+ - unsloth
11
+ - agent
12
+ - tool-use
13
+ - alfworld
14
+ - dbbench
15
  ---
16
 
17
+ # Qwen3-4B Agent Trajectory (v11)
18
+
19
+ This repository provides a **fully merged model** fine-tuned from **Qwen/Qwen3-4B-Instruct-2507** using Unsloth.
20
+
21
+ Unlike standard adapter repositories, this repository contains the **merged weights**, meaning you do not need to load the base model separately.
22
+
23
+ ## Training Objective
24
+
25
+ This model is trained to improve **multi-turn agent task performance**
26
+ on ALFWorld (household tasks) and DBBench (database operations).
27
+
28
+ Loss is applied to **all assistant turns** in the multi-turn trajectory,
29
+ enabling the model to learn environment observation, action selection,
30
+ tool use, and recovery from errors.
31
+
32
+ ## Data Processing
33
+
34
+ - Train/Validation Split: 95% / 5%
35
+ - Random Seed: 3407 (used for shuffling and initialization)
36
+ - Loss Masking: Loss was computed only on the assistant's responses. User prompts and observations were masked during training (`train_on_responses_only` was applied to `<|im_start|>assistant\n`).
37
+
38
+ ## Training Configuration
39
+ - **Base model**: Qwen/Qwen3-4B-Instruct-2507
40
+ - **Method**: LoRA + Unsloth (Merged in 16-bit)
41
+ - **Max sequence length**: 8192
42
+ - **Epochs**: 1
43
+ - **Learning rate**: 5e-06
44
+ - **LoRA**: r=16, alpha=32
45
+ - **PER_DEVICE_TRAIN_BATCH_SIZE** = 4
46
+ - **GRAD_ACCUM** = 2
47
+ - **WARMUP_RATIO** = 0.1
48
+ - **WEIGHT_DECAY** = 0.05
49
+ - **NEFTUNE_NOISE_ALPHA** = 5.0
50
+ - **VAL_RATIO** = 0.05
51
+
52
+ ## Usage
53
+
54
+ ```python
55
+ from transformers import AutoModelForCausalLM, AutoTokenizer
56
+ import torch
57
+
58
+ model_id = "choco800/qwen3-4b-agent-v11"
59
 
60
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
61
+ model = AutoModelForCausalLM.from_pretrained(
62
+ model_id,
63
+ torch_dtype=torch.bfloat16,
64
+ device_map="auto",
65
+ )
66
+ ```
67
+ ## Sources & Terms (IMPORTANT)
68
 
69
+ Training data:
70
+ - u-10bei/dbbench_sft_dataset_react
71
 
72
+ Dataset License: MIT License. These datasets are used and distributed under the terms of the MIT License.
73
+ Compliance: Users must comply with the dataset licenses and the base model's original terms of use (Apache 2.0).