diff --git a/examples/textual_inversion/textual_inversion_sdxl.py b/examples/textual_inversion/textual_inversion_sdxl.py index 90f1b36b08ea..d35a11b70a20 100644 --- a/examples/textual_inversion/textual_inversion_sdxl.py +++ b/examples/textual_inversion/textual_inversion_sdxl.py @@ -546,6 +546,8 @@ def __getitem__(self, i): example["original_size"] = (image.height, image.width) + image = image.resize((self.size, self.size), resample=self.interpolation) + if self.center_crop: y1 = max(0, int(round((image.height - self.size) / 2.0))) x1 = max(0, int(round((image.width - self.size) / 2.0))) @@ -576,7 +578,6 @@ def __getitem__(self, i): img = np.array(image).astype(np.uint8) image = Image.fromarray(img) - image = image.resize((self.size, self.size), resample=self.interpolation) image = self.flip_transform(image) image = np.array(image).astype(np.uint8)