github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Create an apply_monkeypatches() start util.

Browse Source
This commit is contained in:
Ryan Dick
2025-02-21 15:16:40 +00:00
parent ca23b5337e
commit f345c0fabc
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/api_app.py
View File

@@ -13,9 +13,7 @@ from fastapi.responses import HTMLResponse, RedirectResponse
from fastapi_events.handlers.local import local_handler
from fastapi_events.middleware import EventHandlerASGIMiddleware
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
from torch.backends.mps import is_available as is_mps_available
import invokeai.backend.util.hotfixes # noqa: F401 (monkeypatching on import)
import invokeai.frontend.web as web_dir
from invokeai.app.api.dependencies import ApiDependencies
from invokeai.app.api.no_cache_staticfiles import NoCacheStaticFiles
@@ -38,15 +36,13 @@ from invokeai.app.util.custom_openapi import get_openapi_func
# for PyCharm:
# noinspection PyUnresolvedReferences
from invokeai.app.util.startup_utils import check_cudnn, enable_dev_reload, find_open_port
from invokeai.app.util.startup_utils import apply_monkeypatches, check_cudnn, enable_dev_reload, find_open_port
from invokeai.backend.util.devices import TorchDevice
from invokeai.backend.util.logging import InvokeAILogger
app_config = get_config()
if is_mps_available():
import invokeai.backend.util.mps_fixes # noqa: F401 (monkeypatching on import)
apply_monkeypatches()
logger = InvokeAILogger.get_logger(config=app_config)

9
invokeai/app/util/startup_utils.py
View File

@@ -44,3 +44,12 @@ def enable_dev_reload() -> None:
) from e
else:
jurigged.watch(logger=InvokeAILogger.get_logger(name="jurigged").info)
def apply_monkeypatches() -> None:
"""Apply monkeypatches to fix issues with third-party libraries."""
import invokeai.backend.util.hotfixes # noqa: F401 (monkeypatching on import)
if torch.backends.mps.is_available():
import invokeai.backend.util.mps_fixes # noqa: F401 (monkeypatching on import)