add sd-disable-nsfw arg

2022-09-29 09:42:19 +08:00
parent 1a92569f00
commit 0d57e552cf
3 changed files with 41 additions and 1 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -218,6 +218,7 @@ def main(args):
        name=args.model,
        device=device,
        hf_access_token=args.hf_access_token,
+        sd_disable_nsfw=args.sd_disable_nsfw,
        callbacks=[diffuser_callback],
    )