biyHddlmZerddlmZddlmZdddefd Zy ) )is_tqdm_available)tqdm) PartialStateT)main_process_onlyrcts tdt|dkDrt|dtr t d|j dd}|r|stjdk7}t|i|d|iS)z Wrapper around `tqdm.tqdm` that optionally displays only on the main process. Args: main_process_only (`bool`, *optional*): Whether to display the progress bar only on the main process zZAccelerate's `tqdm` module requires `tqdm` to be installed. Please run `pip install tqdm`.rzPassing `True` or `False` as the first argument to Accelerate's `tqdm` wrapper is unsupported. Please use the `main_process_only` keyword argument instead.disableF) r ImportErrorlen isinstancebool ValueErrorpoprlocal_process_index_tqdm)rargskwargsr s P/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/accelerate/utils/tqdm.pyrrs  vww 4y1}DGT2 K  jjE*G.449 $ 2& 2' 22N)importsr tqdm.autorrstaterrrrrs) '' +/343r