Fixed for DFEs with pixelspace and video models

2026-07-27 11:12:02 -06:00
parent 0e17841767
commit fb204b7677
4 changed files with 92 additions and 53 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -796,7 +796,10 @@ class SDTrainer(BaseSDTrainProcess):
                        tv = torch.clamp(tv, min=0.001)
                
                # step latent, use here or with do_fft_loss
-                t0 = noisy_latents - tv * noise_pred
+                if self.sd.x0_pred:
+                    t0 = noise_pred
+                else:
+                    t0 = noisy_latents - tv * noise_pred
                
                if self.train_config.t0_loss_target:
                    # replace the loss targets and pred