Update examples/dreambooth/train_dreambooth_lora_hidream.py

Co-authored-by: Bagheera <[email protected]>

Author: linoytsaban

examples/dreambooth/train_dreambooth_lora_hidream.py

@@ -1662,7 +1662,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
                 # and instead post-weight the loss
                 weighting = compute_loss_weighting_for_sd3(weighting_scheme=args.weighting_scheme, sigmas=sigmas)
 
-                # flow matching loss (reversed)
+                # flow matching loss via reverse ODE (HiDream uses an inverted flow field as compared to the more typical `noise - model_input` target)
                 target = model_input - noise
 
                 if args.with_prior_preservation: