File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff 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"
Original file line number Diff line number Diff 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"
Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments