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

fix: should invoke callback as well in api only mode
This commit is contained in:
AUTOMATIC1111
2022-11-02 13:04:55 +03:00
committed by GitHub
2 changed files with 4 additions and 1 deletions

View File

@@ -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)