Adding Conditioning Blocks to the Diffusion Transformer.

2026-10-18 00:33:52 +00:00 · 2025-01-15 22:29:47 +00:00
parent f34b8659c3
commit b341b79652
7 changed files with 173 additions and 40 deletions
@@ -737,7 +737,8 @@ class Trainer:
                        # Default to flow-matching noise addition
                        noisy_latents = (1.0 - sigmas) * latent_conditions["latents"] + sigmas * noise
                        noisy_latents = noisy_latents.to(latent_conditions["latents"].dtype)
-
+                        
+                    # this is used for whatever reason to pass into the 
                    latent_conditions.update({"noisy_latents": noisy_latents})

                    weights = prepare_loss_weights(