Project
requiredA project is one folder holding the downloaded materials, the corpus, the precomputed shards, checkpoints and the exported GGUF. Everything resumes from that folder after a reboot.
Materials
teacher · student · datasetsCorpus
text promptsExtra local prompt files
one prompt per line, or .jsonl with a "text" fieldThe teacher never sees images, so the corpus is text only. ~1% empty prompts teach the checkpoint's empty CFG prompt. 60k prompts took ~3.7 h of teacher time on an RTX 5090 (4.5 prompts/s).
Precompute
teacher + student▾
The teacher is spread over the CUDA devices and then CPU RAM (accelerate device_map). sequential fills the chosen device up to its memory budget, then the other GPUs, then RAM — on a 5090 + 4050 the whole 30 GB backbone stays on the GPUs. balanced is transformers' default even split: it caps the big card at half the model and offloads the rest to the CPU, where each MoE layer runs its 256 experts eagerly (2.4 vs ~4+ prompts/s). If sequential runs out of VRAM, lower the GPU memory budget by 2–3 GiB instead of switching. Token budget caps batch × (text tokens + 256 query rows).
Training
adapter▾
trainer8 recipe: width 1024, depth 6, 20k steps, batch 32, lr 2e-4, warmup 1k, cosine to 10%, whitened MSE + 0.5·(1−cos) on standardized targets. Judge by val centred cosine (0.965 on the 5090 run, ~55 min); below ~0.9 widen (1536) or deepen before touching the recipe.
Output
f16 GGUFPipeline
no projectOnce training has produced best.pt, the Export GGUF and Evaluate boxes are clickable: they regenerate the GGUF / eval.json from the checkpoint at any time, e.g. after the file was deleted.
Training metrics
Live hardware
Hardware
Environment
Pipeline log
Exported files
Evaluation
Judge by cos_centered / rel_mse: the teacher rows share a large per-dim offset, so plain cosine is ~0.99 even for a zero prediction. cos_rms is what the DiT consumes (per-row RMSNorm). roundtrip_cos compares the exported f16 GGUF against the checkpoint (expect ≥ 0.999).
Use it in ggk
Text-to-image works with every student quantization; editing wants pig_clip at q8_0 or better and prefers the f16 DiT (trainer8 A/B).