Commit 308b5101 authored by AUTOMATIC's avatar AUTOMATIC

fix an unlikely division by 0 error

parent 6620acff
......@@ -67,10 +67,13 @@ def progressapi(req: ProgressRequest):
progress = 0
if shared.state.job_count > 0:
progress += shared.state.job_no / shared.state.job_count
if shared.state.sampling_steps > 0:
progress += 1 / shared.state.job_count * shared.state.sampling_step / shared.state.sampling_steps
job_count, job_no = shared.state.job_count, shared.state.job_no
sampling_steps, sampling_step = shared.state.sampling_steps, shared.state.sampling_step
if job_count > 0:
progress += job_no / job_count
if sampling_steps > 0:
progress += 1 / job_count * sampling_step / sampling_steps
progress = min(progress, 1)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment