Unverified Commit 0c18d02c authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Remove spurious arg in training scripts (#1644)

Remove spurious arg in training scripts.
parent 6b68afd8
...@@ -491,7 +491,7 @@ def main(args): ...@@ -491,7 +491,7 @@ def main(args):
if is_xformers_available(): if is_xformers_available():
try: try:
unet.enable_xformers_memory_efficient_attention(True) unet.enable_xformers_memory_efficient_attention()
except Exception as e: except Exception as e:
logger.warning( logger.warning(
"Could not enable memory efficient attention. Make sure xformers is installed" "Could not enable memory efficient attention. Make sure xformers is installed"
......
...@@ -367,7 +367,7 @@ def main(): ...@@ -367,7 +367,7 @@ def main():
if is_xformers_available(): if is_xformers_available():
try: try:
unet.enable_xformers_memory_efficient_attention(True) unet.enable_xformers_memory_efficient_attention()
except Exception as e: except Exception as e:
logger.warning( logger.warning(
"Could not enable memory efficient attention. Make sure xformers is installed" "Could not enable memory efficient attention. Make sure xformers is installed"
......
...@@ -442,7 +442,7 @@ def main(): ...@@ -442,7 +442,7 @@ def main():
if is_xformers_available(): if is_xformers_available():
try: try:
unet.enable_xformers_memory_efficient_attention(True) unet.enable_xformers_memory_efficient_attention()
except Exception as e: except Exception as e:
logger.warning( logger.warning(
"Could not enable memory efficient attention. Make sure xformers is installed" "Could not enable memory efficient attention. Make sure xformers is installed"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment