diff --git a/download-model.py b/download-model.py index 9ee77906..2642c405 100644 --- a/download-model.py +++ b/download-model.py @@ -23,7 +23,7 @@ from tqdm.contrib.concurrent import thread_map class ModelDownloader: - def __init__(self, max_retries): + def __init__(self, max_retries = 5): self.s = requests.Session() if max_retries: self.s.mount('https://cdn-lfs.huggingface.co', HTTPAdapter(max_retries=max_retries))