Skip to content

Commit ef1e628

Browse files
authored
fix style (#11975)
up
1 parent 173e1b1 commit ef1e628

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1057,7 +1057,7 @@ def __init__(
10571057
if interpolation is None:
10581058
raise ValueError(f"Unsupported interpolation mode {interpolation=}.")
10591059
train_resize = transforms.Resize(size, interpolation=interpolation)
1060-
train_crop = transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size)
1060+
train_crop = transforms.CenterCrop(size) if args.center_crop else transforms.RandomCrop(size)
10611061
train_flip = transforms.RandomHorizontalFlip(p=1.0)
10621062
train_transforms = transforms.Compose(
10631063
[
@@ -1101,7 +1101,7 @@ def __init__(
11011101
self.image_transforms = transforms.Compose(
11021102
[
11031103
transforms.Resize(size, interpolation=interpolation),
1104-
transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size),
1104+
transforms.CenterCrop(size) if args.center_crop else transforms.RandomCrop(size),
11051105
transforms.ToTensor(),
11061106
transforms.Normalize([0.5], [0.5]),
11071107
]

0 commit comments

Comments
 (0)