Views
No views yet
Qwen/Qwen3-0.6B, trained with TRL SFT on a small sample of trl-lib/Capybara.Qwen/Qwen3-0.6BSFTTrainertrl-lib/Capybaratransformers, peft, trl, torch| Metric | Value |
|---|---|
| Train samples | 8 |
| Eval samples | 2 |
| Max steps | 1 |
| Max length | 256 |
| Train loss | 1.1917 |
| Baseline eval loss | 2.4017 |
| Final eval loss | 2.2526 |
pip install transformers peft accelerate torch1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
4base_model = "Qwen/Qwen3-0.6B"
5adapter_id = "edgemindroboticslabs/qwen3-0.6b-capybara-sft"
6
7tokenizer = AutoTokenizer.from_pretrained(adapter_id)
8model = AutoModelForCausalLM.from_pretrained(base_model, device_map="auto")
9model = PeftModel.from_pretrained(model, adapter_id)
10
11pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
12messages = [{"role": "user", "content": "Give me three practical tips for organizing a small robotics lab."}]
13print(pipe(messages, max_new_tokens=160, do_sample=True, temperature=0.7)[0]["generated_text"])uv run --python /opt/homebrew/bin/python3.11 scripts/local_train_sft.py1BASE_MODEL=Qwen/Qwen3-0.6B
2DATASET_ID=trl-lib/Capybara
3MAX_TRAIN_SAMPLES=8
4MAX_EVAL_SAMPLES=2
5MAX_STEPS=1
6MAX_LENGTH=256