Add argument for enabling xFormers optimizations

2022-11-28 17:34:22 -08:00
parent 0d2d1ab4ec
commit a0240721e3
4 changed files with 11 additions and 1 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -256,6 +256,7 @@ def main(args):
        sd_disable_nsfw=args.sd_disable_nsfw,
        sd_cpu_textencoder=args.sd_cpu_textencoder,
        sd_run_local=args.sd_run_local,
+        sd_enable_xformers=args.sd_enable_xformers,
        callback=diffuser_callback,
    )