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

Bug Fix

This commit is contained in:
cafeai 2022-12-01 05:10:21 +09:00
parent fb75cbe029
commit 1074bd6f3b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trainer/diffusers_trainer.py
View File

@ -932,7 +932,7 @@ def main():
if global_step % args.image_log_steps == 0: if global_step % args.image_log_steps == 0:
if rank == 0: if rank == 0:
# get prompt from random batch # get prompt from random batch
prompt = tokenizer.decode(batch['tokens'][random.randint(0, len(batch['tokens'])-1)].tolist()) prompt = tokenizer.decode(batch['tokens'][random.randint(0, len(batch['tokens'])-1)])
if args.image_log_scheduler == 'DDIMScheduler': if args.image_log_scheduler == 'DDIMScheduler':
print('using DDIMScheduler scheduler') print('using DDIMScheduler scheduler')