Skip to content

Commit 0236a91

Browse files
committed
Added LANCZOS as default interplotation mode
1 parent e94b91d commit 0236a91

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

examples/controlnet/train_controlnet_sdxl.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
134134

135135
for validation_prompt, validation_image in zip(validation_prompts, validation_images):
136136
validation_image = Image.open(validation_image).convert("RGB")
137-
137+
138138
try:
139139
interpolation = getattr(transforms.InterpolationMode, args.image_interpolation_mode.upper())
140140
except (AttributeError, KeyError):
@@ -145,11 +145,13 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
145145
f"Interpolation mode {args.image_interpolation_mode} is not supported. "
146146
f"Please select one of the following: {', '.join(supported_interpolation_modes)}"
147147
)
148-
149-
transform = transforms.Compose([
150-
transforms.Resize(args.resolution, interpolation=interpolation),
151-
transforms.CenterCrop(args.resolution),
152-
])
148+
149+
transform = transforms.Compose(
150+
[
151+
transforms.Resize(args.resolution, interpolation=interpolation),
152+
transforms.CenterCrop(args.resolution),
153+
]
154+
)
153155
validation_image = transform(validation_image)
154156

155157
images = []
@@ -767,7 +769,7 @@ def prepare_train_dataset(dataset, accelerator):
767769
f"Interpolation mode {args.image_interpolation_mode} is not supported. "
768770
f"Please select one of the following: {', '.join(supported_interpolation_modes)}"
769771
)
770-
772+
771773
image_transforms = transforms.Compose(
772774
[
773775
transforms.Resize(args.resolution, interpolation=interpolation_mode),

0 commit comments

Comments
 (0)