From 45db049973df865cd63f4127057cc820842aadaf Mon Sep 17 00:00:00 2001 From: linjiapro Date: Wed, 18 Oct 2023 13:38:52 -0700 Subject: [PATCH] Fix the order of width and height of original size in SDXL training script (#5382) * wip * wip --------- Co-authored-by: Sayak Paul --- examples/text_to_image/train_text_to_image_lora_sdxl.py | 2 +- examples/text_to_image/train_text_to_image_sdxl.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/text_to_image/train_text_to_image_lora_sdxl.py b/examples/text_to_image/train_text_to_image_lora_sdxl.py index 249b9d1a9a..35de6eedca 100644 --- a/examples/text_to_image/train_text_to_image_lora_sdxl.py +++ b/examples/text_to_image/train_text_to_image_lora_sdxl.py @@ -839,7 +839,7 @@ def main(args): all_images = [] crop_top_lefts = [] for image in images: - original_sizes.append((image.height, image.width)) + original_sizes.append((image.width, image.height)) image = train_resize(image) if args.center_crop: y1 = max(0, int(round((image.height - args.resolution) / 2.0))) diff --git a/examples/text_to_image/train_text_to_image_sdxl.py b/examples/text_to_image/train_text_to_image_sdxl.py index 4a3048a0ba..ea8ceff395 100644 --- a/examples/text_to_image/train_text_to_image_sdxl.py +++ b/examples/text_to_image/train_text_to_image_sdxl.py @@ -825,7 +825,7 @@ def main(args): all_images = [] crop_top_lefts = [] for image in images: - original_sizes.append((image.height, image.width)) + original_sizes.append((image.width, image.height)) image = train_resize(image) if args.center_crop: y1 = max(0, int(round((image.height - args.resolution) / 2.0)))