Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add np seed 

Just in case!
This commit is contained in:
cafeai 2022-12-02 01:49:12 +09:00
parent d225cd57e7
commit c4b17e41f1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trainer/diffusers_trainer.py
View File

@ -718,6 +718,7 @@ def main():
# Set seed # Set seed
torch.manual_seed(args.seed) torch.manual_seed(args.seed)
random.seed(args.seed) random.seed(args.seed)
np.random.seed(args.seed)
print('RANDOM SEED:', args.seed) print('RANDOM SEED:', args.seed)
if args.resume: if args.resume: