P/j<ddlZddlZddlmZmZdZdZejdZes ddl Z dZ ee dd Z d Z y#e $rdZ YwxYw) N) _FaketqdmtqdmTc~tjdtjdtjdgS)Nz%torch.fx.experimental.symbolic_shapesz torch._dynamoztorch._inductor)logging getLoggerX/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/_dynamo/logging.py get_loggersr s8AB/*+, r ztorch.compile())totaldescdelayctsItjdttts tj j ttfd}|S)Nr c2j|d|fi|y)Nz Step %s: %s)log)levelmsgkwargsloggersteps r rzget_step_logger..log8s 5-s=f=r ) disable_progresspbarupdate isinstancerset_postfix_strnamenext _step_counter)rrrs` @r get_step_loggerr"0sE  A$ *  FKK= 2  D> Jr ) itertoolsr torch.hubrrrr countr!triton num_steps ImportErrorrr"rr r r)sk%& "   i&7q AD   sAA  A