o )iy@sBddlmZerddlmZddlmZdddefd d Zd S) )is_tqdm_available)tqdm) PartialStateT)main_process_onlyrcOsftstdt|dkrt|dtrtd|dd}|r(|s(tjdk}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)rargskwargsrrR/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/accelerate/utils/tqdm.pyrs  rN)importsr tqdm.autorrstaterr rrrrs