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

Merge pull request #34 from chavinlo/patch-1 

typo
This commit is contained in:
Anthony Mercurio 2022-11-10 09:23:10 -07:00 committed by GitHub
parent 548ebea881 c4b1860282
commit bc626e80e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
diffusers_trainer.py
View File

@ -521,7 +521,7 @@ def main():
os.makedirs(args.output_path, exist_ok=True) os.makedirs(args.output_path, exist_ok=True)
run = wandb.init(project=args.project_id, name=args.run_name, config=vars(args), dir=args.output_path+'/wandb') run = wandb.init(project=args.project_id, name=args.run_name, config=vars(args), dir=args.output_path+'/wandb')
# Inform the user of host, and various versions -- useful for debugging isseus. # Inform the user of host, and various versions -- useful for debugging issues.
print("RUN_NAME:", args.run_name) print("RUN_NAME:", args.run_name)
print("HOST:", socket.gethostname()) print("HOST:", socket.gethostname())
print("CUDA:", torch.version.cuda) print("CUDA:", torch.version.cuda)