diff --git a/globals.py b/globals.py index e8eb46f..d2bee06 100644 --- a/globals.py +++ b/globals.py @@ -28,6 +28,7 @@ class StreamingPrompt(BaseModel): status_endpoint: Optional[str] file_upload_endpoint: Optional[str] workflow: Any + gpu_event_id: Optional[str] = None class SimplePrompt(BaseModel):