apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

send empty progress with initial job

This commit is contained in:
Sean Sube 2024-01-06 20:32:38 -06:00
parent be40797e59
commit fd92d52339
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/chain/edit_metadata.py
View File

@ -1,6 +1,6 @@
from typing import Any, Optional from typing import Any, Optional
from ..params import HighresParams, ImageParams, Size, StageParams, UpscaleParams from ..params import HighresParams, ImageParams, Size, SizeChart, StageParams, UpscaleParams
from ..server import ServerContext from ..server import ServerContext
from ..worker import WorkerContext from ..worker import WorkerContext
from .base import BaseStage from .base import BaseStage
@ -8,6 +8,8 @@ from .result import StageResult
class EditMetadataStage(BaseStage): class EditMetadataStage(BaseStage):
max_tile = SizeChart.max
def run( def run(
self, self,
_worker: WorkerContext, _worker: WorkerContext,

5
api/onnx_web/server/api.py
View File

@ -97,6 +97,11 @@ def job_reply(name: str):
return jsonify( return jsonify(
{ {
"name": name, "name": name,
"queue": Progress(0, 0).tojson(), # TODO: use real queue position
"status": JobStatus.PENDING,
"stages": Progress(0, 0).tojson(),
"steps": Progress(0, 0).tojson(),
"tiles": Progress(0, 0).tojson(),
} }
) )