Unverified Commit ec932399 authored by MinJu-Ha's avatar MinJu-Ha Committed by GitHub
Browse files

[train_dreambooth_lora_sdxl] Add --image_interpolation_mode option for image...

[train_dreambooth_lora_sdxl] Add --image_interpolation_mode option for image resizing (default to lanczos) (#11490)

feat(train_dreambooth_lora_sdxl): support --image_interpolation_mode with default to lanczos
parent fc5e9066
......@@ -852,7 +852,7 @@ class DreamBoothDataset(Dataset):
self.image_transforms = transforms.Compose(
[
transforms.Resize(size, interpolation=transforms.InterpolationMode.BILINEAR),
transforms.Resize(size, interpolation=interpolation),
transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size),
transforms.ToTensor(),
transforms.Normalize([0.5], [0.5]),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment