choco800 commited on
Commit
bda7753
·
verified ·
1 Parent(s): 956f1b2

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +63 -12
README.md CHANGED
@@ -1,21 +1,72 @@
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/sft_alfworld_trajectory_dataset_v5
 
 
 
 
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
  ---
15
 
16
+ # Qwen3-4B Agent Trajectory (v5)
17
+
18
+ This repository provides a **fully merged model** fine-tuned from **Qwen/Qwen3-4B-Instruct-2507** using Unsloth.
19
+
20
+ Unlike standard adapter repositories, this repository contains the **merged weights**, meaning you do not need to load the base model separately.
21
+
22
+ ## Training Objective
23
+
24
+ This model is trained to improve **multi-turn agent task performance**
25
+ on ALFWorld (household tasks).
26
+
27
+ Loss is applied to **all assistant turns** in the multi-turn trajectory,
28
+ enabling the model to learn environment observation, action selection,
29
+ tool use, and recovery from errors.
30
+
31
+ ## Data Processing
32
+
33
+ - Train/Validation Split: 95% / 5%
34
+ - Random Seed: 3407 (used for shuffling and initialization)
35
+ - 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`).
36
+
37
+ ## Training Configuration
38
+ - **Base model**: Qwen/Qwen3-4B-Instruct-2507
39
+ - **Method**: LoRA + Unsloth (Merged in 16-bit)
40
+ - **Max sequence length**: 8192
41
+ - **Epochs**: 1
42
+ - **Learning rate**: 1e-05
43
+ - **LoRA**: r=16, alpha=32
44
+ - **PER_DEVICE_TRAIN_BATCH_SIZE** = 8
45
+ - **GRAD_ACCUM** = 2
46
+ - **WARMUP_RATIO** = 0.1
47
+ - **WEIGHT_DECAY** = 0.05
48
+ - **NEFTUNE_NOISE_ALPHA** = 5.0
49
+ - **VAL_RATIO** = 0.05
50
+
51
+ ## Usage
52
+
53
+ ```python
54
+ from transformers import AutoModelForCausalLM, AutoTokenizer
55
+ import torch
56
+
57
+ model_id = "choco800/qwen3-4b-agent-v5"
58
 
59
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
60
+ model = AutoModelForCausalLM.from_pretrained(
61
+ model_id,
62
+ torch_dtype=torch.bfloat16,
63
+ device_map="auto",
64
+ )
65
+ ```
66
+ ## Sources & Terms (IMPORTANT)
67
 
68
+ Training data:
69
+ - u-10bei/sft_alfworld_trajectory_dataset_v5
70
 
71
+ Dataset License: MIT License. These datasets are used and distributed under the terms of the MIT License.
72
+ Compliance: Users must comply with the dataset licenses and the base model's original terms of use (Apache 2.0).