All LoRA adapter weights have been merged into the base model.
The resulting checkpoint is fully standalone and does not require
external adapter loading.
The two ALFWorld trajectory datasets were merged and filtered.
-
Admissible Action Requirement
Only samples containing at least one user turn with:
admissible actions:
were retained.
-
Token length was computed using the base model tokenizer.
-
Samples exceeding:
2048 - 148 = 1900 tokens
were discarded.
The two DBBench datasets were merged and rebalanced by task type.
-
Fully retained:
-
aggregation*
-
comparison*
-
counting*
-
ranking*
-
Capped to fixed limits:
-
INSERT (max 200)
-
UPDATE (max 250)
-
INSERT_error_recovery (max 60)
-
UPDATE_error_recovery (max 40)
-
Other task types:
-
Randomly sampled up to a total of 120 examples.
After sampling, the dataset was globally shuffled.
No explicit task-type prefix (e.g., "Task Type: DATABASE") was inserted.
Original ReAct-style conversational structure was preserved.
The model is optimized for multi-turn reasoning and action-based tasks.
Loss is applied to all assistant turns in multi-turn trajectories,
allowing the model to learn:
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "Umiharu/Qwen-4B-DB-AlfWorld-v17"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.float16,
10 device_map="auto",
11)
12
13prompt = "You are a task-solving agent. Respond 'OK' if you are ready."
14inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
15
16outputs = model.generate(
17 **inputs,
18 max_new_tokens=64,
19 temperature=0.2,
20 do_sample=False,
21)
22
23print(tokenizer.decode(outputs[0], skip_special_tokens=True))
24
25## Sources & Terms (IMPORTANT)
26
27Training data: u-10bei/sft_alfworld_trajectory_dataset_v4, u-10bei/sft_alfworld_trajectory_dataset_v5, u-10bei/dbbench_sft_dataset_react_v2, u-10bei/dbbench_sft_dataset_react_v3
28
29Dataset License: MIT License. This dataset is used and distributed under the terms of the MIT License.
30Compliance: Users must comply with the MIT license (including copyright notice) and the base model's original terms of use.