handle MaxRetryError

2024-01-08 23:43:20 +08:00
parent 7c1f83e71d
commit 6f4ce66793
2 changed files with 9 additions and 1 deletions
--- a/iopaint/model_manager.py
+++ b/iopaint/model_manager.py
@@ -39,7 +39,7 @@ class ModelManager:
        logger.info(f"Loading model: {name}")
        if name not in self.available_models:
            raise NotImplementedError(
-                f"Unsupported model: {name}. Available models: {self.available_models.keys()}"
+                f"Unsupported model: {name}. Available models: {list(self.available_models.keys())}"
            )

        model_info = self.available_models[name]