From c03bd2917c35b44d6a38a63dac4b0f72dad3ab99 Mon Sep 17 00:00:00 2001 From: MrZhengXin <34998703+MrZhengXin@users.noreply.github.com> Date: Wed, 31 Aug 2022 20:02:22 +0800 Subject: [PATCH] enable http request retry under proxy (#1022) * enable http request retry under proxy * Update src/huggingface_hub/file_download.py Co-authored-by: zhengxin.1999 Co-authored-by: Lucain --- src/huggingface_hub/file_download.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/huggingface_hub/file_download.py b/src/huggingface_hub/file_download.py index e988cbd0e8..f6eaf0f802 100644 --- a/src/huggingface_hub/file_download.py +++ b/src/huggingface_hub/file_download.py @@ -463,7 +463,10 @@ def _request_wrapper( method=method.upper(), url=url, timeout=timeout, **params ) success = True - except requests.exceptions.ConnectTimeout as err: + except ( + requests.exceptions.ConnectTimeout, + requests.exceptions.ProxyError, + ) as err: if tries > max_retries: raise err else: