diff --git a/comfy/cmd/server.py b/comfy/cmd/server.py index 6d3580c36..fe972a044 100644 --- a/comfy/cmd/server.py +++ b/comfy/cmd/server.py @@ -25,7 +25,7 @@ from aiohttp import web from can_ada import URL, parse as urlparse # pylint: disable=no-name-in-module from typing_extensions import NamedTuple -from model_filemanager import download_model, DownloadModelStatus +from ..model_filemanager import download_model, DownloadModelStatus from .latent_preview_image_encoding import encode_preview_image from .. import interruption from .. import model_management diff --git a/model_filemanager/__init__.py b/comfy/model_filemanager/__init__.py similarity index 100% rename from model_filemanager/__init__.py rename to comfy/model_filemanager/__init__.py diff --git a/model_filemanager/download_models.py b/comfy/model_filemanager/download_models.py similarity index 100% rename from model_filemanager/download_models.py rename to comfy/model_filemanager/download_models.py