Allow random noise shift with video latents.

2026-07-31 05:58:08 -06:00
parent ddc69745fe
commit 6e7bc81241
1 changed files with 1 additions and 1 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -1368,7 +1368,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
                if self.train_config.random_noise_shift > 0.0:
                    # get random noise -1 to 1
                    noise_shift = torch.randn(
-                        batch_size, latents.shape[1], 1, 1,
+                        s,
                        device=noise.device,
                        dtype=noise.dtype
                    ) * self.train_config.random_noise_shift