Fix crop trigger size not having a default value
This commit is contained in:
parent
c3157b4e15
commit
785848b5ec
2
main.py
2
main.py
@ -102,7 +102,7 @@ def get_args_parser():
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--port", default=8080, type=int)
|
||||
parser.add_argument("--model", default="lama", choices=["lama", "ldm"])
|
||||
parser.add_argument("--crop-trigger-size", nargs=2, type=int,
|
||||
parser.add_argument("--crop-trigger-size", default=[2042, 2042], nargs=2, type=int,
|
||||
help="If image size large then crop-trigger-size, "
|
||||
"crop each area from original image to do inference."
|
||||
"Mainly for performance and memory reasons"
|
||||
|
Loading…
Reference in New Issue
Block a user