Instructions to use burtenshaw/tmax2-rawcmd-tmaxsft-lora-lr1e5-r8-s80 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use burtenshaw/tmax2-rawcmd-tmaxsft-lora-lr1e5-r8-s80 with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("allenai/tmax-2b") model = PeftModel.from_pretrained(base_model, "burtenshaw/tmax2-rawcmd-tmaxsft-lora-lr1e5-r8-s80") - Notebooks
- Google Colab
- Kaggle
TMax2 Raw-Command TMax-SFT LoRA
This is a LoRA adapter for allenai/tmax-2b trained as a terminal-agent raw-command distillation rung.
Method
- Base model:
allenai/tmax-2b - Method: TRL SFT with PEFT LoRA, rank 8, learning rate
1e-5, 80 steps - Training data:
allenai/tmax-sft, configskill_tax_20260505_2.2k_combined_balanced_thinking_only_success - Conversion: successful teacher traces were converted into prompt/completion rows where the completion is only the next bash command
- Data size: 384 train episodes, 3,581 train action rows; 64 held-out episodes, 619 held-out action rows
- Training job: SLURM
22303226, onehopper-prodnode,qos=low - Trackio: https://huggingface.co/spaces/burtenshaw/terminal-bench-loop-trackio
Proxy Evaluation
This is not a full Terminal-Bench result. It is a raw-command TBLite proxy over four held-out tasks: log-summary, jsonl-aggregator, protein-sequence, and schedule-vacation.
| Model | Strict proxy success | Valid command rate | Avg turns | Notes |
|---|---|---|---|---|
allenai/tmax-2b base |
0/4 | 0.7083 | 6.0 | Same evaluator run |
| This adapter | 1/4 | 0.9000 | 5.0 | Solved jsonl-aggregator |
The adapter improved over the same-size base on this proxy gate, but it does not approach the final project goal of Terminal-Bench > 40. Treat it as a promotion candidate for the next controlled rung, not a benchmark win.
Artifacts
- Training root:
/fsx/benjamin_burtenshaw/post-training-agent-experiments/2026-06-24-tmax2-rawcmd-trackio-sft/ - Eval summary in this repo:
eval/tblite_rawcmd_proxy/eval_summary.json - Combined base-vs-adapter summary:
eval/tblite_rawcmd_proxy/combined_summary.json
- Downloads last month
- 88
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support