add sd-cpu-textencoder args

2022-09-29 12:20:55 +08:00
parent 0d57e552cf
commit ec7b2d8e2d
3 changed files with 14 additions and 4 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -219,6 +219,7 @@ def main(args):
        device=device,
        hf_access_token=args.hf_access_token,
        sd_disable_nsfw=args.sd_disable_nsfw,
+        sd_cpu_textencoder=args.sd_cpu_textencoder,
        callbacks=[diffuser_callback],
    )