diff options
Diffstat (limited to 'modules/script_callbacks.py')
| -rw-r--r-- | modules/script_callbacks.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py index ec1469d0..54824582 100644 --- a/modules/script_callbacks.py +++ b/modules/script_callbacks.py @@ -5,7 +5,7 @@ from typing import Optional, Dict, Any from fastapi import FastAPI
from gradio import Blocks
-from modules import errors
+from modules import errors, timer
def report_exception(c, job):
@@ -123,6 +123,7 @@ def app_started_callback(demo: Optional[Blocks], app: FastAPI): for c in callback_map['callbacks_app_started']:
try:
c.callback(demo, app)
+ timer.startup_timer.record(c.script)
except Exception:
report_exception(c, 'app_started_callback')
|