From c5d7baec799c6c0dd0385a391db7fd8111c74f0e Mon Sep 17 00:00:00 2001 From: Qing Date: Sun, 4 Sep 2022 15:20:07 +0800 Subject: [PATCH] add fcf to args --- lama_cleaner/parse_args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lama_cleaner/parse_args.py b/lama_cleaner/parse_args.py index 4cad49d..6481f82 100644 --- a/lama_cleaner/parse_args.py +++ b/lama_cleaner/parse_args.py @@ -7,7 +7,7 @@ def parse_args(): parser = argparse.ArgumentParser() parser.add_argument("--host", default="127.0.0.1") parser.add_argument("--port", default=8080, type=int) - parser.add_argument("--model", default="lama", choices=["lama", "ldm", "zits", "mat"]) + parser.add_argument("--model", default="lama", choices=["lama", "ldm", "zits", "mat", 'fcf']) parser.add_argument("--device", default="cuda", type=str, choices=["cuda", "cpu"]) parser.add_argument("--gui", action="store_true", help="Launch as desktop app") parser.add_argument(