fix cv2 params

2022-10-09 21:32:13 +08:00
parent 521a1e2858
commit f4fcece180
6 changed files with 37 additions and 1 deletions
--- a/lama_cleaner/schema.py
+++ b/lama_cleaner/schema.py
@@ -44,3 +44,7 @@ class Config(BaseModel):
    sd_sampler: str = SDSampler.ddim
    # -1 mean random seed
    sd_seed: int = 42
+
+    # cv2
+    cv2_flag: str = 'INPAINT_NS'
+    cv2_radius: int = 4