From d225cd57e7aa1d16211883ee44293fd6f7772f13 Mon Sep 17 00:00:00 2001
From: cafeai <116491182+cafeai@users.noreply.github.com>
Date: Fri, 2 Dec 2022 01:41:22 +0900
Subject: [PATCH] Fix Seed

---
 trainer/diffusers_trainer.py | 1 +
 1 file changed, 1 insertion(+)

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: