diff --git a/transformerlab/models/ollamamodel.py b/transformerlab/models/ollamamodel.py index 13952471e..924370372 100644 --- a/transformerlab/models/ollamamodel.py +++ b/transformerlab/models/ollamamodel.py @@ -13,6 +13,9 @@ async def list_models(): try: ollama_model_library = ollama_models_library_dir() + except FileNotFoundError: + print("Skipping Ollama models: manifests directory not found") + return [] except Exception as e: print("Failed to locate Ollama models library:") print(str(e)) diff --git a/transformerlab/routers/tasks.py b/transformerlab/routers/tasks.py index 9e972fa19..b0cadc015 100644 --- a/transformerlab/routers/tasks.py +++ b/transformerlab/routers/tasks.py @@ -619,8 +619,6 @@ async def get_task_file_content(task_dir: str, file_path: str): return {"status": "error", "message": "An error occurred while getting task file content"} - - @router.post("/install_from_gallery", summary="Install a task from transformerlab/galleries to local tasks-gallery") async def install_task_from_gallery( id: str = Form(...), @@ -958,9 +956,7 @@ async def import_task_from_local_gallery( remote_info = resp.json() # Extract uploaded_dir path from response uploaded_dir = ( - remote_info.get("uploaded_files", {}) - .get("dir_files", {}) - .get("uploaded_dir") + remote_info.get("uploaded_files", {}).get("dir_files", {}).get("uploaded_dir") ) if uploaded_dir: try: