ghostai1 commited on
Commit
4266e41
·
verified ·
1 Parent(s): 86aaa4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -930,8 +930,9 @@ def worker(img_np, prompt, negative_prompt, seed, secs, win, stp, cfg, gsc, rsc,
930
  pv = einops.rearrange(pv, "b c t h w -> (b h) (t w) c")
931
  cur = d["i"] + 1
932
  job_status[jid]["progress"] = (cur / stp) * 100
933
- logger.info(f"Job {jid} Progress {cur}/{stp} ({job_status[jid]['progress']:.1f}%)")
934
- print(f"{yellow(f'API: Job {jid} Progress {cur}/{stp} ({job_status[jid]['progress']:.1f}%)')}")
 
935
  stream.output_queue.push(('progress', (pv, f"{cur}/{stp}", make_progress_bar_html(int(100 * cur / stp), f"{cur}/{stp}"))))
936
  try:
937
  if hasattr(stream.input_queue, "qsize") and stream.input_queue.qsize() > 0:
 
930
  pv = einops.rearrange(pv, "b c t h w -> (b h) (t w) c")
931
  cur = d["i"] + 1
932
  job_status[jid]["progress"] = (cur / stp) * 100
933
+ progress_message = f"API: Job {jid} Progress {cur}/{stp} ({job_status[jid]['progress']:.1f}%)"
934
+ logger.info(progress_message)
935
+ print(yellow(progress_message))
936
  stream.output_queue.push(('progress', (pv, f"{cur}/{stp}", make_progress_bar_html(int(100 * cur / stp), f"{cur}/{stp}"))))
937
  try:
938
  if hasattr(stream.input_queue, "qsize") and stream.input_queue.qsize() > 0: