remove explicit disable xformers

This commit is contained in:
Victor Hall 2023-01-02 18:38:08 -05:00
parent 07125135a8
commit a7dcf43f7c
1 changed files with 1 additions and 1 deletions

View File

@ -392,11 +392,11 @@ def main(args):
if not args.disable_xformers and is_xformers_available():
try:
unet.enable_xformers_memory_efficient_attention()
logging.info("Enabled xformers")
except Exception as ex:
logging.warning("failed to load xformers, continuing without it")
pass
else:
unet.disable_xformers_memory_efficient_attention()
logging.info("xformers not available or disabled")
default_lr = 3e-6