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/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -235,6 +235,9 @@ class StableDiffusion:
        # if a mask is passed, do the loss with the mask. May be set false for models that use a mask for other reasons.
        self.do_masked_loss = True
        
+        # if the model outputs an x0 prediction (clean latent)
+        self.x0_pred = False
+        
    # properties for old arch for backwards compatibility
    @property
    def is_xl(self):