add run-sd-local arg

2022-09-29 13:13:09 +08:00
parent ec7b2d8e2d
commit 4d6cec0ca2
4 changed files with 22 additions and 13 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -220,6 +220,7 @@ def main(args):
        hf_access_token=args.hf_access_token,
        sd_disable_nsfw=args.sd_disable_nsfw,
        sd_cpu_textencoder=args.sd_cpu_textencoder,
+        sd_run_local=args.sd_run_local,
        callbacks=[diffuser_callback],
    )