From 1074bd6f3b3ccf18d5cd60f7efe7c9a7b00f45d3 Mon Sep 17 00:00:00 2001 From: cafeai <116491182+cafeai@users.noreply.github.com> Date: Thu, 1 Dec 2022 05:10:21 +0900 Subject: [PATCH] Bug Fix --- trainer/diffusers_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py index f65f3cb..2c31014 100644 --- a/trainer/diffusers_trainer.py +++ b/trainer/diffusers_trainer.py @@ -932,7 +932,7 @@ def main(): if global_step % args.image_log_steps == 0: if rank == 0: # 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': print('using DDIMScheduler scheduler')