mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-07-11 17:52:04 +00:00
25 lines
772 B
Python
25 lines
772 B
Python
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()
|