Add np seed

Just in case!
2022-12-02 01:49:12 +09:00 · 2022-12-02 01:49:12 +09:00 · c4b17e41f1
parent d225cd57e7
commit c4b17e41f1
1 changed files with 1 additions and 0 deletions
--- 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: