Spaces:
Running on Zero
Running on Zero
| import importlib | |
| import logging | |
| def force_hf_hub_progress_bars(): | |
| hf_tqdm = importlib.import_module('huggingface_hub.utils.tqdm') | |
| original_is_tqdm_disabled = hf_tqdm.is_tqdm_disabled | |
| if getattr(original_is_tqdm_disabled, '_aitk_forced_progress', False): | |
| return | |
| def is_tqdm_disabled(log_level): | |
| disabled = original_is_tqdm_disabled(log_level) | |
| # UI jobs log to files, so stderr is not a TTY. Keep HF download bars | |
| # visible while preserving normal carriage-return progress updates. | |
| if disabled is None and log_level != logging.NOTSET: | |
| return False | |
| return disabled | |
| is_tqdm_disabled._aitk_forced_progress = True | |
| hf_tqdm.is_tqdm_disabled = is_tqdm_disabled | |
| force_hf_hub_progress_bars() | |