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
0c18d02c
Unverified
Commit
0c18d02c
authored
Dec 10, 2022
by
Pedro Cuenca
Committed by
GitHub
Dec 10, 2022
Browse files
Remove spurious arg in training scripts (#1644)
Remove spurious arg in training scripts.
parent
6b68afd8
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
examples/dreambooth/train_dreambooth.py
examples/dreambooth/train_dreambooth.py
+1
-1
examples/text_to_image/train_text_to_image.py
examples/text_to_image/train_text_to_image.py
+1
-1
examples/textual_inversion/textual_inversion.py
examples/textual_inversion/textual_inversion.py
+1
-1
No files found.
examples/dreambooth/train_dreambooth.py
View file @
0c18d02c
...
...
@@ -491,7 +491,7 @@ def main(args):
if
is_xformers_available
():
try
:
unet
.
enable_xformers_memory_efficient_attention
(
True
)
unet
.
enable_xformers_memory_efficient_attention
()
except
Exception
as
e
:
logger
.
warning
(
"Could not enable memory efficient attention. Make sure xformers is installed"
...
...
examples/text_to_image/train_text_to_image.py
View file @
0c18d02c
...
...
@@ -367,7 +367,7 @@ def main():
if
is_xformers_available
():
try
:
unet
.
enable_xformers_memory_efficient_attention
(
True
)
unet
.
enable_xformers_memory_efficient_attention
()
except
Exception
as
e
:
logger
.
warning
(
"Could not enable memory efficient attention. Make sure xformers is installed"
...
...
examples/textual_inversion/textual_inversion.py
View file @
0c18d02c
...
...
@@ -442,7 +442,7 @@ def main():
if
is_xformers_available
():
try
:
unet
.
enable_xformers_memory_efficient_attention
(
True
)
unet
.
enable_xformers_memory_efficient_attention
()
except
Exception
as
e
:
logger
.
warning
(
"Could not enable memory efficient attention. Make sure xformers is installed"
...
...
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