FlowRead / test_tqdm.py
jobbler's picture
feat: Add percentage progress to downloading status in UI and extension
dbffc3c
raw
history blame contribute delete
605 Bytes
import huggingface_hub.utils as hf_utils
original_tqdm = hf_utils.tqdm
class MyTqdm(original_tqdm):
def update(self, n=1):
super().update(n)
if hasattr(self, 'total') and self.total:
print(f"DL_PROGRESS: {self.n / self.total:.2%}", flush=True)
hf_utils.tqdm = MyTqdm
hf_utils.tqdm.tqdm = MyTqdm # Also overwrite the module attribute just in case
import huggingface_hub.file_download
huggingface_hub.file_download.tqdm = MyTqdm
try:
from transformers import AutoConfig
AutoConfig.from_pretrained("gpt2", force_download=True)
except Exception as e:
print(e)