From c4b17e41f1dde5c4d4cda5b6dd7e957fc353db9b Mon Sep 17 00:00:00 2001
From: cafeai <116491182+cafeai@users.noreply.github.com>
Date: Fri, 2 Dec 2022 01:49:12 +0900
Subject: [PATCH] Add np seed

Just in case!
---
 trainer/diffusers_trainer.py | 1 +
 1 file changed, 1 insertion(+)

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