Skip to content

Commit 0c18d02

Browse files
authored
Remove spurious arg in training scripts (#1644)
Remove spurious arg in training scripts.
1 parent 6b68afd commit 0c18d02

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

examples/dreambooth/train_dreambooth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ def main(args):
491491

492492
if is_xformers_available():
493493
try:
494-
unet.enable_xformers_memory_efficient_attention(True)
494+
unet.enable_xformers_memory_efficient_attention()
495495
except Exception as e:
496496
logger.warning(
497497
"Could not enable memory efficient attention. Make sure xformers is installed"

examples/text_to_image/train_text_to_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,7 @@ def main():
367367

368368
if is_xformers_available():
369369
try:
370-
unet.enable_xformers_memory_efficient_attention(True)
370+
unet.enable_xformers_memory_efficient_attention()
371371
except Exception as e:
372372
logger.warning(
373373
"Could not enable memory efficient attention. Make sure xformers is installed"

examples/textual_inversion/textual_inversion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,7 @@ def main():
442442

443443
if is_xformers_available():
444444
try:
445-
unet.enable_xformers_memory_efficient_attention(True)
445+
unet.enable_xformers_memory_efficient_attention()
446446
except Exception as e:
447447
logger.warning(
448448
"Could not enable memory efficient attention. Make sure xformers is installed"

0 commit comments

Comments
 (0)