kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

add fcf to args

Browse Source
This commit is contained in:
Qing 2022-09-04 15:20:07 +08:00
parent 9ccc577be2
commit c5d7baec79
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/parse_args.py
View File

@ -7,7 +7,7 @@ def parse_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--host", default="127.0.0.1") parser.add_argument("--host", default="127.0.0.1")
parser.add_argument("--port", default=8080, type=int) 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("--device", default="cuda", type=str, choices=["cuda", "cpu"])
parser.add_argument("--gui", action="store_true", help="Launch as desktop app") parser.add_argument("--gui", action="store_true", help="Launch as desktop app")
parser.add_argument( parser.add_argument(