1.1.1

2023-04-06 22:56:19 +08:00
parent d531d3efbb
commit 2c25cff9f7
2 changed files with 4 additions and 1 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -140,10 +140,13 @@ class Config(BaseModel):
    output_dir: str = None
    # plugins
    enable_interactive_seg: bool = False
+    interactive_seg_model: str = "vit_l"
+    interactive_seg_device: str = "cuda"
    enable_remove_bg: bool = False
    enable_realesrgan: bool = False
    realesrgan_device: str = "cpu"
    realesrgan_model: str = RealESRGANModelName.realesr_general_x4v3.value
+    realesrgan_no_half: bool = False
    enable_gfpgan: bool = False
    gfpgan_device: str = "cpu"
    enable_restoreformer: bool = False