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
df55f053
Unverified
Commit
df55f053
authored
May 30, 2025
by
Justin Ruan
Committed by
GitHub
May 29, 2025
Browse files
Fix wrong indent for examples of controlnet script (#11632)
fix wrong indent for training controlnet
parent
89ddb6c0
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
16 additions
and
16 deletions
+16
-16
examples/controlnet/train_controlnet.py
examples/controlnet/train_controlnet.py
+4
-4
examples/controlnet/train_controlnet_flux.py
examples/controlnet/train_controlnet_flux.py
+3
-3
examples/controlnet/train_controlnet_sd3.py
examples/controlnet/train_controlnet_sd3.py
+5
-5
examples/controlnet/train_controlnet_sdxl.py
examples/controlnet/train_controlnet_sdxl.py
+4
-4
No files found.
examples/controlnet/train_controlnet.py
View file @
df55f053
...
@@ -178,11 +178,11 @@ def log_validation(
...
@@ -178,11 +178,11 @@ def log_validation(
else
:
else
:
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
del
pipeline
del
pipeline
gc
.
collect
()
gc
.
collect
()
torch
.
cuda
.
empty_cache
()
torch
.
cuda
.
empty_cache
()
return
image_logs
return
image_logs
def
import_model_class_from_model_name_or_path
(
pretrained_model_name_or_path
:
str
,
revision
:
str
):
def
import_model_class_from_model_name_or_path
(
pretrained_model_name_or_path
:
str
,
revision
:
str
):
...
...
examples/controlnet/train_controlnet_flux.py
View file @
df55f053
...
@@ -192,9 +192,9 @@ def log_validation(
...
@@ -192,9 +192,9 @@ def log_validation(
else
:
else
:
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
del
pipeline
del
pipeline
free_memory
()
free_memory
()
return
image_logs
return
image_logs
def
save_model_card
(
repo_id
:
str
,
image_logs
=
None
,
base_model
=
str
,
repo_folder
=
None
):
def
save_model_card
(
repo_id
:
str
,
image_logs
=
None
,
base_model
=
str
,
repo_folder
=
None
):
...
...
examples/controlnet/train_controlnet_sd3.py
View file @
df55f053
...
@@ -199,13 +199,13 @@ def log_validation(controlnet, args, accelerator, weight_dtype, step, is_final_v
...
@@ -199,13 +199,13 @@ def log_validation(controlnet, args, accelerator, weight_dtype, step, is_final_v
else
:
else
:
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
del
pipeline
del
pipeline
free_memory
()
free_memory
()
if
not
is_final_validation
:
if
not
is_final_validation
:
controlnet
.
to
(
accelerator
.
device
)
controlnet
.
to
(
accelerator
.
device
)
return
image_logs
return
image_logs
# Copied from dreambooth sd3 example
# Copied from dreambooth sd3 example
...
...
examples/controlnet/train_controlnet_sdxl.py
View file @
df55f053
...
@@ -201,11 +201,11 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
...
@@ -201,11 +201,11 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
else
:
else
:
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
logger
.
warning
(
f
"image logging not implemented for
{
tracker
.
name
}
"
)
del
pipeline
del
pipeline
gc
.
collect
()
gc
.
collect
()
torch
.
cuda
.
empty_cache
()
torch
.
cuda
.
empty_cache
()
return
image_logs
return
image_logs
def
import_model_class_from_model_name_or_path
(
def
import_model_class_from_model_name_or_path
(
...
...
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