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

MinJu-Ha · web-flow · commit ec9323996b62 · 2025-05-05T10:19:30.000-04:00
feat(train_dreambooth_lora_sdxl): support --image_interpolation_mode with default to lanczos
diff --git a/examples/dreambooth/train_dreambooth_lora_sdxl.py b/examples/dreambooth/train_dreambooth_lora_sdxl.py
@@ -852,7 +852,7 @@ def __init__(
 
         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]),

Original file line number	Diff line number	Diff line change
`@@ -852,7 +852,7 @@ def __init__(`
`852`	`852`
`853`	`853`	`self.image_transforms = transforms.Compose(`
`854`	`854`	`[`
`855`		`- transforms.Resize(size, interpolation=transforms.InterpolationMode.BILINEAR),`
	`855`	`+ transforms.Resize(size, interpolation=interpolation),`
`856`	`856`	`transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size),`
`857`	`857`	`transforms.ToTensor(),`
`858`	`858`	`transforms.Normalize([0.5], [0.5]),`