This adapter is a self-distillation experiment, not a general-purpose coding model release. It uses TRL's experimental SDPO trainer with include_environment_feedback=True, so filtered trace diagnostics are supplied as privileged_context for teacher-conditioned reprompts.

Training Run

Field	Value
Hub repo	`burtenshaw/gemma-4-12b-sdpo-pi-mono-trace-feedback-v3`
Base model	`google/gemma-4-12B-it`
Dataset	`badlogicgames/pi-mono`
Dataset split	`train`
Trackio SDK	`static`
Selected samples	`128`
Minimum filter score	`9.6`
Average filter score	`13.289`
Max steps	`64`
Learning rate	`3e-05`
Training method	`4-bit NF4 QLoRA + TRL SDPO`
LoRA rank	`8`
LoRA alpha	`16`
Num generations	`4`
Success reward threshold	`0.45`
Max prompt length	`768`
Max completion length	`160`

Trackio: Trackio run
Run name: gemma4-12b-sdpo-pi-mono-static-r64-g4-lr3e5-th45-20260604
Output directory: outputs/sdpo-pi-mono-trace-feedback

Data Filtering

The source data is badlogicgames/pi-mono. The script uses the Dataset Viewer parquet export by default because the raw JSONL files have schema drift that can make direct load_dataset() reconstruction brittle.

A row is kept when it has a substantive prompt plus at least one concrete tool error, environment diagnostic, or later user correction. Rows are scored higher for test/build failures, runtime exceptions, missing files or commands, explicit user feedback, and evidence that the trace continued after the failure.

Selected-data summary:

Source sessions: 128
Source files: 128
Score range: 9.6 to 19.5
Average score: 13.289

Category counts:

build_lint_compile: 62
command_error: 127
missing_file_or_command: 69
other_marked_error: 5
permission_auth: 3
runtime_exception: 30
test_or_assertion: 112
tool_schema_validation: 5
user_feedback: 111

First selected sample preview:

text

score: 19.5
categories: build_lint_compile, command_error, missing_file_or_command, runtime_exception, test_or_assertion, user_feedback
reward_terms: build_lint_compile, command_error, missing_file_or_command, runtime_exception, test_or_assertion, user_feedback, environment, diagnostic, promises:332, triggeruncaughtexception

prompt:
Analyze GitHub issue(s): https://github.com/badlogic/pi-mono/issues/2291

For each issue:

1. Read the issue in full, including all comments and linked issues/PRs.
2. Do not trust analysis written in the issue. Independently verify behavior and derive your own analysis from the code and execution path.

3. **For bugs**:
   - Ignore any root cause analysis in the issue (likely wrong)
   - Read all related code files in full (no truncation)
   - Trace the code path and identify the actual root cause
   - Propose a fix

4. **For feature requests**:
   - Do not trust implementation proposals in the issue without verification
   - Read all related code files in full (no truncation)
   - Propose the most concise implementation approach
   - List affected files and changes needed

Do NOT implement unless explicitly asked. Analyze and propose only.

privileged_context:
Tool/environment diagnostic 1 (test_or_assertion, runtime_exception, command_error):
node:internal/process/promises:332
    triggerUncaughtException(err, true /* fromPromise */);
    ^

Error: Transform failed with 3 errors:
/eval.ts:29:2: ERROR: Top-level await is currently not supported with the "cjs" output format
/eval.ts:31:22: ERROR: Top-level await is currently not supported with the "cjs" output format
/eval.ts:46:2: ERROR: Top-level await is currently not supported with the "cjs" output format
    at failureErrorWithLog ($WORKSPACE/node_modules/esbuild/lib/main.js:1748:15)
    at $WOR

[... trimmed ...]

el' does not exist in type 'SimpleStreamOptions'.
packages/coding-agent/src/core/extensions/runner.ts(242,46): error TS2304: Cannot find name 'ProviderConfig'.

Command exited with code 2

Tool/environment diagnostic 3 (test_or_assertion):
No changes made to packages/coding-agent/src/core/agent-session.ts. The replacement produced identical content. This might indicate an issue with special characters or the text not existing as expected.

Later user correction 1:
what'st he most concise fix? this sounds overly complex

Later user correction 2:
well, we can't just fix it for session_start then

Reward Function

The included trace_grounding_reward is intentionally lightweight. It rewards completions that look like concrete coding-agent responses and mention terms grounded in the trace diagnostic. That is enough to exercise SDPO, Trackio, HF Jobs, LoRA push-to-Hub, and the filtered trace format.

For a serious training run, replace the heuristic reward with a verifier that can replay or grade the task, such as tests, build checks, lint output, tool-call validation, or another sandboxed outcome signal.

Pi Harness via VLLM

This repository contains a PEFT LoRA adapter, not a merged full model. A practical harness setup is to serve the Gemma base model once with VLLM, attach this adapter as a named LoRA module, and point the Pi harness at VLLM's OpenAI-compatible /v1 endpoint.

bash

python -m pip install -U "vllm>=0.10.0" "huggingface_hub>=0.30.0"

hf download burtenshaw/gemma-4-12b-sdpo-pi-mono-trace-feedback-v3 \
  --repo-type model \
  --local-dir ./adapters/pi-mono-sdpo

vllm serve google/gemma-4-12B-it \
  --host 0.0.0.0 \
  --port 8000 \
  --dtype bfloat16 \
  --api-key token-pi-harness \
  --enable-lora \
  --lora-modules pi-mono-sdpo=./adapters/pi-mono-sdpo

Then configure the Pi harness as an OpenAI-compatible provider:

bash

export OPENAI_BASE_URL=http://127.0.0.1:8000/v1
export OPENAI_API_KEY=token-pi-harness
export OPENAI_MODEL=pi-mono-sdpo

Smoke-test the server before running the harness:

bash

curl http://127.0.0.1:8000/v1/chat/completions \
  -H "Authorization: Bearer token-pi-harness" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "pi-mono-sdpo",
    "messages": [
      {"role": "system", "content": "You are a careful coding agent."},
      {"role": "user", "content": "A test says expected 2 but got 3. What should you inspect first?"}
    ],
    "max_tokens": 128
  }'

If your Pi harness uses different variable names, map the same three values: OpenAI-compatible base URL, API key, and model name. If your VLLM version cannot serve this Gemma 4 adapter directly, merge the adapter into the base model first or run the adapter with a Transformers/PEFT inference process behind the same OpenAI-compatible API.

Training Metrics

epoch: 0.25
total_flos: 0.0
train_loss: -0.3233257165993564
train_runtime: 1459.0617
train_samples_per_second: 0.088
train_steps_per_second: 0.044

Reproduce

bash

hf jobs uv run https://gist.githubusercontent.com/burtenshaw/61c4a4b367409bf7a6000cc0ec54c483/raw/8bd586d69b3367a2aa8ffd6abcdcaefa4e8ab9af/train_sdpo_pi_mono_full.py \
  --flavor a10g-large \
  --timeout 5h \
  --secrets HF_TOKEN \
  --env HUB_MODEL_ID=burtenshaw/gemma-4-12b-sdpo-pi-mono-trace-feedback-v3 \
  --mode train \
  --max-samples 128 \
  --max-steps 64 \
  --num-generations 4 \
  --learning-rate 3e-5 \
  --success-reward-threshold 0.45 \
  --trackio-space-id burtenshaw/sdpo-pi-mono-trackio-static-v4 \
  --trackio-sdk static \
  --run-name gemma4-12b-sdpo-pi-mono-static-r64-g4-lr3e5-th45-20260604

Limitations

The data is filtered from existing traces, so it reflects the trace collector's task distribution and failure modes.
The reward is a smoke-test heuristic and should not be interpreted as a reliable coding benchmark.
The model is pushed as a PEFT LoRA adapter; load it with the base model listed above.
SDPO is experimental in TRL, so pin versions for long-running comparisons.