Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
ef1e6287
Unverified
Commit
ef1e6287
authored
Jul 22, 2025
by
YiYi Xu
Committed by
GitHub
Jul 22, 2025
Browse files
fix style (#11975)
up
parent
173e1b14
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py
...diffusion_training/train_dreambooth_lora_flux_advanced.py
+2
-2
No files found.
examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py
View file @
ef1e6287
...
@@ -1057,7 +1057,7 @@ class DreamBoothDataset(Dataset):
...
@@ -1057,7 +1057,7 @@ class DreamBoothDataset(Dataset):
if
interpolation
is
None
:
if
interpolation
is
None
:
raise
ValueError
(
f
"Unsupported interpolation mode
{
interpolation
=
}
."
)
raise
ValueError
(
f
"Unsupported interpolation mode
{
interpolation
=
}
."
)
train_resize
=
transforms
.
Resize
(
size
,
interpolation
=
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_flip
=
transforms
.
RandomHorizontalFlip
(
p
=
1.0
)
train_transforms
=
transforms
.
Compose
(
train_transforms
=
transforms
.
Compose
(
[
[
...
@@ -1101,7 +1101,7 @@ class DreamBoothDataset(Dataset):
...
@@ -1101,7 +1101,7 @@ class DreamBoothDataset(Dataset):
self
.
image_transforms
=
transforms
.
Compose
(
self
.
image_transforms
=
transforms
.
Compose
(
[
[
transforms
.
Resize
(
size
,
interpolation
=
interpolation
),
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
.
ToTensor
(),
transforms
.
Normalize
([
0.5
],
[
0.5
]),
transforms
.
Normalize
([
0.5
],
[
0.5
]),
]
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment