diff --git a/model_filemanager/model_filemanager.py b/app/model_manager.py similarity index 100% rename from model_filemanager/model_filemanager.py rename to app/model_manager.py diff --git a/model_filemanager/__init__.py b/model_filemanager/__init__.py index f5d0c7bbc..b7ac16256 100644 --- a/model_filemanager/__init__.py +++ b/model_filemanager/__init__.py @@ -1,3 +1,2 @@ # model_manager/__init__.py from .download_models import download_model, DownloadModelStatus, DownloadStatusType, create_model_path, check_file_exists, track_download_progress, validate_filename -from .model_filemanager import ModelFileManager \ No newline at end of file diff --git a/server.py b/server.py index e4b20a75c..e68c38ed9 100644 --- a/server.py +++ b/server.py @@ -29,7 +29,8 @@ import comfy.model_management import node_helpers from app.frontend_management import FrontendManager from app.user_manager import UserManager -from model_filemanager import download_model, DownloadModelStatus, ModelFileManager +from app.model_manager import ModelFileManager +from model_filemanager import download_model, DownloadModelStatus from typing import Optional from api_server.routes.internal.internal_routes import InternalRoutes