change default interactive_seg_device to cpu;add interactive seg model/device to web_config

2023-04-16 09:41:35 +08:00
parent 6750df13e7
commit d3f6f69af6
2 changed files with 19 additions and 3 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -141,7 +141,7 @@ class Config(BaseModel):
    # plugins
    enable_interactive_seg: bool = False
    interactive_seg_model: str = "vit_l"
-    interactive_seg_device: str = "cuda"
+    interactive_seg_device: str = "cpu"
    enable_remove_bg: bool = False
    enable_realesrgan: bool = False
    realesrgan_device: str = "cpu"