Skip to content

0.18.0: epochs, validation pack, held-out photos, Fizgig target - #4

Merged
EnragedAntelope merged 1 commit into
mainfrom
feat/0.18.0-training-export
Oct 8, 2026
Merged

EnragedAntelope merged 1 commit into
mainfrom
feat/0.18.0-training-export

Conversation

@EnragedAntelope

Copy link
Copy Markdown
Owner

Phase 1 of the LoRA best-practice update: training export.

  • ⑤ counts epochs (one checkpoint + one sample set each). Repeats 0 = auto, sized to the n × 75 step target; Fizgig auto = 1. A live exposure line shows the total.
  • Every config writes a validation/ pack: 8 fixed prompts at seed 42, validation.md (how to pick the epoch, inference settings, held-out photos), and a per-epoch score sheet.
  • ai-toolkit saves and samples once per epoch and keeps every epoch. musubi, kohya and Fizgig sample every epoch from the prompt file. musubi Krea 2 samples via --turbo_dit.
  • New Fizgig target: Krea 2, Qwen Image 2.1 (704), Klein. Commands follow Fizgig's docs/CLI.md, with <<FILL>> model paths.
  • ① stamps a dd_source provenance PNG chunk. ⑤ names the sources that were upscaled, and ④ records real-photo provenance in metadata.json.
  • ④ adds Hold out N reference photos (and cli export --holdout). The largest real photos are copied to <dataset>-heldout/, never into the training folder.

Deferred to Future ideas: the optimizer dropdown and the attention-only toggle.

Tests: pytest -q (616 passed), ruff check . clean.

- ⑤ counts epochs (one checkpoint + one sample set each); Repeats 0 = auto, sized to n x 75 target steps; Fizgig auto = 1
- ⑤ live exposure line (images x repeats / batch = steps/epoch x epochs) with low/high warnings
- every config writes validation/: 8 fixed prompts at seed 42, validation.md, per-epoch score sheet
- ai-toolkit saves/samples once per epoch, keeps every epoch, seed 42 with walk_seed off
- musubi/kohya/Fizgig sample every epoch from the prompt file; musubi Krea 2 samples via --turbo_dit
- Fizgig trainer target (Krea 2, Qwen Image 2.1 at 704, Klein) with its cache/train commands
- ① writes a dd_source provenance PNG chunk; ⑤ names upscaled sources; ④ records provenance in metadata.json
- ④ "Hold out N reference photos" (+ cli export --holdout) copies the largest real photos to <dataset>-heldout/
- dataset naming skips names whose -heldout sibling exists
- version 0.18.0; AGENTS.md, ARCHITECTURE.md, README updated
@EnragedAntelope
EnragedAntelope merged commit b95cb7f into main Oct 8, 2026
3 checks passed
@EnragedAntelope
EnragedAntelope deleted the feat/0.18.0-training-export branch October 8, 2026 16:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant