1580158
/
sd_dreambooth_extension
mirror of https://github.com/d8ahazard/sd_dreambooth_extension.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Respect --api flag when adding API endpoints

pull/1392/head
d8ahazard 2023-11-18 10:32:48 -06:00
parent a6f28dcc7e
commit 18afe4f394
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/api.py
View File

@ -934,10 +934,14 @@ def dreambooth_api(_, app: FastAPI):
try:
import modules.script_callbacks as script_callbacks
from modules.shared import cmd_opts
if cmd_opts.api:
import modules.script_callbacks as script_callbacks
script_callbacks.on_app_started(dreambooth_api)
logger.debug("SD-Webui API layer loaded")
script_callbacks.on_app_started(dreambooth_api)
logger.debug("SD-Webui API layer loaded")
else:
logger.debug("API flag not enabled, skipping API layer. Please enable with --api")
except:
logger.debug("Unable to import script callbacks.")
pass