From 6e7bc812410e085c2ef3344b28ad6d410e4241f2 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Fri, 31 Jul 2026 05:58:08 -0600 Subject: [PATCH] Allow random noise shift with video latents. --- jobs/process/BaseSDTrainProcess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jobs/process/BaseSDTrainProcess.py b/jobs/process/BaseSDTrainProcess.py index 94228a5..c73a621 100644 --- 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