Unverified Commit e6060a7e authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #4155 from MaikoTan/fix/register-api-in-api-only-mode

fix: should invoke callback as well in api only mode
parents eb5e82c7 dd2108fd
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -2,6 +2,7 @@ import sys
import traceback
from collections import namedtuple
import inspect
from typing import Optional

from fastapi import FastAPI
from gradio import Blocks
@@ -62,7 +63,7 @@ def clear_callbacks():
    callbacks_image_saved.clear()
    callbacks_cfg_denoiser.clear()

def app_started_callback(demo: Blocks, app: FastAPI):
def app_started_callback(demo: Optional[Blocks], app: FastAPI):
    for c in callbacks_app_started:
        try:
            c.callback(demo, app)
+2 −0
Original line number Diff line number Diff line
@@ -114,6 +114,8 @@ def api_only():
    app.add_middleware(GZipMiddleware, minimum_size=1000)
    api = create_api(app)

    modules.script_callbacks.app_started_callback(None, app)

    api.launch(server_name="0.0.0.0" if cmd_opts.listen else "127.0.0.1", port=cmd_opts.port if cmd_opts.port else 7861)