Spaces:
Running on Zero
Running on Zero
File size: 772 Bytes
855e1f5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | 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()
|