update

2023-12-28 10:48:52 +08:00
parent f0b852725f
commit 9cc9bd7a88
19 changed files with 345 additions and 482 deletions
--- a/lama_cleaner/model_manager.py
+++ b/lama_cleaner/model_manager.py
@@ -37,7 +37,7 @@ class ModelManager:
    def init_model(self, name: str, device, **kwargs):
        logger.info(f"Loading model: {name}")
        if name not in self.available_models:
-            raise NotImplementedError(f"Unsupported model: {name}")
+            raise NotImplementedError(f"Unsupported model: {name}. Available models: {self.available_models.keys()}")

        model_info = self.available_models[name]
        kwargs = {