diff --git a/lama_cleaner/const.py b/lama_cleaner/const.py index 64d4018..7fd8dac 100644 --- 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 diff --git a/setup.py b/setup.py index a37a2ee..e1c4dd4 100644 --- a/setup.py +++ b/setup.py @@ -21,7 +21,7 @@ def load_requirements(): # https://setuptools.readthedocs.io/en/latest/setuptools.html#including-data-files setuptools.setup( name="lama-cleaner", - version="1.1.0", + version="1.1.1", author="PanicByte", author_email="cwq1913@gmail.com", description="Image inpainting tool powered by SOTA AI Model",