add --no-half arg

2023-01-03 21:30:33 +08:00
parent 6cfc7c30f1
commit 59ee89bd34
5 changed files with 6 additions and 3 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -303,6 +303,7 @@ def main(args):
    model = ModelManager(
        name=args.model,
        device=device,
+        no_half=args.no_half,
        hf_access_token=args.hf_access_token,
        sd_disable_nsfw=args.sd_disable_nsfw,
        sd_cpu_textencoder=args.sd_cpu_textencoder,