From ec7016e97a276ddd2e703a5aca862c387651d3f4 Mon Sep 17 00:00:00 2001 From: linjia Date: Thu, 12 Oct 2023 17:20:03 -0700 Subject: [PATCH 1/2] wip --- examples/text_to_image/train_text_to_image_sdxl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 c681943f2e94..4a2ff1370df6 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 preprocess_train(examples): 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))) From cde93b77b7f6f04ef1f33703d8b019e281053593 Mon Sep 17 00:00:00 2001 From: linjia Date: Mon, 16 Oct 2023 11:40:05 -0700 Subject: [PATCH 2/2] wip --- examples/text_to_image/train_text_to_image_lora_sdxl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 249b9d1a9ab5..35de6eedcabd 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 preprocess_train(examples): 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)))