Skip to content

Commit bd1d9bc

Browse files
authored
Add ZeroDivisionError catch for EasyCache logging statement (#9768)
1 parent fb763d4 commit bd1d9bc

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

comfy_extras/nodes_easycache.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,12 @@ def easycache_sample_wrapper(executor, *args, **kwargs):
162162
logging.info(f"{easycache.name} [verbose] - output_change_rates {len(output_change_rates)}: {output_change_rates}")
163163
logging.info(f"{easycache.name} [verbose] - approx_output_change_rates {len(approx_output_change_rates)}: {approx_output_change_rates}")
164164
total_steps = len(args[3])-1
165-
logging.info(f"{easycache.name} - skipped {easycache.total_steps_skipped}/{total_steps} steps ({total_steps/(total_steps-easycache.total_steps_skipped):.2f}x speedup).")
165+
# catch division by zero for log statement; sucks to crash after all sampling is done
166+
try:
167+
speedup = total_steps/(total_steps-easycache.total_steps_skipped)
168+
except ZeroDivisionError:
169+
speedup = 1.0
170+
logging.info(f"{easycache.name} - skipped {easycache.total_steps_skipped}/{total_steps} steps ({speedup:.2f}x speedup).")
166171
easycache.reset()
167172
guider.model_options = orig_model_options
168173

0 commit comments

Comments
 (0)