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):
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"
......
......@@ -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"
......
......@@ -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"
......
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