From 82814a7048a18d3a27a8a039cd062bd2282c9ca5 Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Tue, 22 Jul 2025 21:34:11 +0200 Subject: [PATCH] up --- .../train_dreambooth_lora_flux_advanced.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py index f3d2e93ea526..0b2e721b941f 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py @@ -1057,7 +1057,7 @@ def __init__( if interpolation is None: raise ValueError(f"Unsupported interpolation mode {interpolation=}.") train_resize = transforms.Resize(size, interpolation=interpolation) - train_crop = transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size) + train_crop = transforms.CenterCrop(size) if args.center_crop else transforms.RandomCrop(size) train_flip = transforms.RandomHorizontalFlip(p=1.0) train_transforms = transforms.Compose( [ @@ -1101,7 +1101,7 @@ def __init__( self.image_transforms = transforms.Compose( [ transforms.Resize(size, interpolation=interpolation), - transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size), + transforms.CenterCrop(size) if args.center_crop else transforms.RandomCrop(size), transforms.ToTensor(), transforms.Normalize([0.5], [0.5]), ]