diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index c8cdb66..4d4e426 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -1047,6 +1047,8 @@ def add_api_endpoints(_: gr.Blocks, app: FastAPI): @app.get("/adetailer/v1/schema") async def schema(): + if hasattr(ADetailerArgs, "model_json_schema"): + return ADetailerArgs.model_json_schema() return ADetailerArgs.schema() @app.get("/adetailer/v1/ad_model")