apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

initialize fields

This commit is contained in:
Sean Sube 2024-01-03 21:20:46 -06:00
parent 0c504e3f69
commit 4f1bc84fd9
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/pipeline.py
View File

@ -33,6 +33,8 @@ class ChainProgress:
self.parent = parent self.parent = parent
self.step = start self.step = start
self.total = 0 self.total = 0
self.stage = 0
self.tile = 0
def __call__(self, step: int, timestep: int, latents: Any) -> None: def __call__(self, step: int, timestep: int, latents: Any) -> None:
if step < self.step: if step < self.step:

1
api/onnx_web/worker/context.py
View File

@ -57,6 +57,7 @@ class WorkerContext:
self.initial_retries = retries self.initial_retries = retries
self.retries = retries self.retries = retries
self.timeout = timeout self.timeout = timeout
self.callback = None
def start(self, job: JobCommand) -> None: def start(self, job: JobCommand) -> None:
# set job name and type # set job name and type