diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py
index f1ad9a0..407303a 100644
--- a/trainer/diffusers_trainer.py
+++ b/trainer/diffusers_trainer.py
@@ -717,6 +717,7 @@ def main():
 
     # Set seed
     torch.manual_seed(args.seed)
+    random.seed(args.seed)
     print('RANDOM SEED:', args.seed)
 
     if args.resume: