mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-09-10 11:35:40 +00:00
Add ZeroDivisionError catch for EasyCache logging statement (#9768)
This commit is contained in:
@@ -162,7 +162,12 @@ def easycache_sample_wrapper(executor, *args, **kwargs):
|
|||||||
logging.info(f"{easycache.name} [verbose] - output_change_rates {len(output_change_rates)}: {output_change_rates}")
|
logging.info(f"{easycache.name} [verbose] - output_change_rates {len(output_change_rates)}: {output_change_rates}")
|
||||||
logging.info(f"{easycache.name} [verbose] - approx_output_change_rates {len(approx_output_change_rates)}: {approx_output_change_rates}")
|
logging.info(f"{easycache.name} [verbose] - approx_output_change_rates {len(approx_output_change_rates)}: {approx_output_change_rates}")
|
||||||
total_steps = len(args[3])-1
|
total_steps = len(args[3])-1
|
||||||
logging.info(f"{easycache.name} - skipped {easycache.total_steps_skipped}/{total_steps} steps ({total_steps/(total_steps-easycache.total_steps_skipped):.2f}x speedup).")
|
# catch division by zero for log statement; sucks to crash after all sampling is done
|
||||||
|
try:
|
||||||
|
speedup = total_steps/(total_steps-easycache.total_steps_skipped)
|
||||||
|
except ZeroDivisionError:
|
||||||
|
speedup = 1.0
|
||||||
|
logging.info(f"{easycache.name} - skipped {easycache.total_steps_skipped}/{total_steps} steps ({speedup:.2f}x speedup).")
|
||||||
easycache.reset()
|
easycache.reset()
|
||||||
guider.model_options = orig_model_options
|
guider.model_options = orig_model_options
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user