update

lama_cleaner/plugins/interactive_seg.py

@@ -22,7 +22,7 @@ SEGMENT_ANYTHING_MODELS = {
         "url": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
         "md5": "4b8939a88964f0f4ff5f5b2642c598a6",
     },
-    "vit_t": {
+    "mobile_sam": {
         "url": "https://github.com/Sanster/models/releases/download/MobileSAM/mobile_sam.pt",
         "md5": "f3c0d8cda613564d499310dab6c812cd",
     },

lama_cleaner/plugins/realesrgan.py

@@ -3,7 +3,7 @@ from enum import Enum
 import cv2
 from loguru import logger
 
-from lama_cleaner.const import RealESRGANModelName
+from lama_cleaner.const import RealESRGANModel
 from lama_cleaner.helper import download_model
 from lama_cleaner.plugins.base_plugin import BasePlugin
 
@@ -18,7 +18,7 @@ class RealESRGANUpscaler(BasePlugin):
         from realesrgan.archs.srvgg_arch import SRVGGNetCompact
 
         REAL_ESRGAN_MODELS = {
-            RealESRGANModelName.realesr_general_x4v3: {
+            RealESRGANModel.realesr_general_x4v3: {
                 "url": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth",
                 "scale": 4,
                 "model": lambda: SRVGGNetCompact(
@@ -31,7 +31,7 @@ class RealESRGANUpscaler(BasePlugin):
                 ),
                 "model_md5": "91a7644643c884ee00737db24e478156",
             },
-            RealESRGANModelName.RealESRGAN_x4plus: {
+            RealESRGANModel.RealESRGAN_x4plus: {
                 "url": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth",
                 "scale": 4,
                 "model": lambda: RRDBNet(
@@ -44,7 +44,7 @@ class RealESRGANUpscaler(BasePlugin):
                 ),
                 "model_md5": "99ec365d4afad750833258a1a24f44ca",
             },
-            RealESRGANModelName.RealESRGAN_x4plus_anime_6B: {
+            RealESRGANModel.RealESRGAN_x4plus_anime_6B: {
                 "url": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth",
                 "scale": 4,
                 "model": lambda: RRDBNet(

lama_cleaner/plugins/segment_anything/build_sam.py

@@ -109,7 +109,7 @@ sam_model_registry = {
     "vit_h": build_sam,
     "vit_l": build_sam_vit_l,
     "vit_b": build_sam_vit_b,
-    "vit_t": build_sam_vit_t,
+    "mobile_sam": build_sam_vit_t,
 }