From 785848b5ec448bc9fedaec9fbee94bc46ecbd85e Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Fri, 25 Mar 2022 05:56:00 +1300
Subject: [PATCH] Fix crop trigger size not having a default value

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 7424b02..67eda09 100644
--- a/main.py
+++ b/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"