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

Fix Seed

This commit is contained in:
cafeai 2022-12-02 01:41:22 +09:00
parent b96eabfe2a
commit d225cd57e7
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

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