fix res for sample

This commit is contained in:
Victor Hall 2022-12-20 13:51:42 -05:00
parent 28da46bee5
commit 1bfe6e97fb
1 changed files with 1 additions and 1 deletions

View File

@ -667,7 +667,7 @@ def main(args):
else: else:
max_prompts = min(4,len(batch["captions"])) max_prompts = min(4,len(batch["captions"]))
prompts=batch["captions"][:max_prompts] prompts=batch["captions"][:max_prompts]
__generate_test_samples(pipe=pipe, prompts=prompts, log_writer=log_writer, log_folder=log_folder, gs=global_step, random_captions=True) __generate_test_samples(pipe=pipe, prompts=prompts, log_writer=log_writer, log_folder=log_folder, gs=global_step, random_captions=True, resolution=args.resolution)
del pipe del pipe
torch.cuda.empty_cache() torch.cuda.empty_cache()