From 2ebba9488b3096335a9beaa6e806552f93dfa211 Mon Sep 17 00:00:00 2001 From: Gnome Ann <> Date: Wed, 11 May 2022 15:51:48 -0400 Subject: [PATCH] Change `force_download` back to False This is to prevent fully downloaded models from being re-downloaded in Colab. --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils.py b/utils.py index 12876f6f..f65222a4 100644 --- a/utils.py +++ b/utils.py @@ -135,7 +135,7 @@ def decodenewlines(txt): #==================================================================# # Downloads sharded huggingface checkpoints using aria2c if possible #==================================================================# -def aria2_hook(pretrained_model_name_or_path: str, force_download=True, cache_dir=None, proxies=None, resume_download=False, local_files_only=False, use_auth_token=None, user_agent=None, revision=None, mirror=None, **kwargs): +def aria2_hook(pretrained_model_name_or_path: str, force_download=False, cache_dir=None, proxies=None, resume_download=False, local_files_only=False, use_auth_token=None, user_agent=None, revision=None, mirror=None, **kwargs): import transformers import transformers.modeling_utils from huggingface_hub import HfFolder