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): ...@@ -852,7 +852,7 @@ class DreamBoothDataset(Dataset):
self.image_transforms = transforms.Compose( 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.CenterCrop(size) if center_crop else transforms.RandomCrop(size),
transforms.ToTensor(), transforms.ToTensor(),
transforms.Normalize([0.5], [0.5]), 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