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()