From ea0c9c951fab37bcffb3dcb28044c151f5849461 Mon Sep 17 00:00:00 2001 From: MinjuHa Date: Mon, 5 May 2025 17:06:22 +0900 Subject: [PATCH] feat(train_dreambooth_lora_sdxl): support --image_interpolation_mode with default to lanczos --- examples/dreambooth/train_dreambooth_lora_sdxl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth_lora_sdxl.py b/examples/dreambooth/train_dreambooth_lora_sdxl.py index 90979ee8ff1c..8af8020bb3fc 100644 --- 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]),