Add sample_start_step setting to configure when sampling starts (#949)

Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
2026-07-15 12:15:50 -05:00
parent 4ece17b71f
commit 8bbd051667
31 changed files with 46 additions and 2 deletions
--- a/config/examples/train_lora_hidream_48.yaml
+++ b/config/examples/train_lora_hidream_48.yaml
@@ -86,6 +86,7 @@ config:
      sample:
        sampler: "flowmatch" # must match train.noise_scheduler
        sample_every: 250 # sample every this many steps
+        sample_start_step: 0 # start sampling at this step
        width: 1024
        height: 1024
        prompts: