From 7e9e25cf6c3bca2a7a3009b5a511ba284232b187 Mon Sep 17 00:00:00 2001 From: Aryan Date: Thu, 12 Sep 2024 13:04:31 +0530 Subject: [PATCH] Update examples/cogvideo/train_cogvideox_lora.py Co-authored-by: YiYi Xu --- examples/cogvideo/train_cogvideox_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/cogvideo/train_cogvideox_lora.py b/examples/cogvideo/train_cogvideox_lora.py index ba7bc27bad..3c9d313425 100644 --- a/examples/cogvideo/train_cogvideox_lora.py +++ b/examples/cogvideo/train_cogvideox_lora.py @@ -1306,7 +1306,7 @@ def main(args): ) # Sample noise that will be added to the latents - noise = torch.rand_like(model_input) + noise = torch.randn_like(model_input) batch_size, num_frames, num_channels, height, width = model_input.shape # Sample a random timestep for each image