MNT Update HF Hub download kwargs (huggingface#2492)

Author: BenjaminBossan

src/peft/config.py

@@ -19,7 +19,7 @@
 from typing import Optional, Union
 
 from huggingface_hub import hf_hub_download
-from transformers.utils import PushToHubMixin
+from transformers.utils import PushToHubMixin, http_user_agent
 
 from .utils import CONFIG_NAME, PeftType, TaskType
 
@@ -190,6 +190,8 @@ def from_pretrained(cls, pretrained_model_name_or_path: str, subfolder: Optional
         )
 
         hf_hub_download_kwargs, class_kwargs, _ = cls._split_kwargs(kwargs)
+        if "user_agent" not in hf_hub_download_kwargs:
+            hf_hub_download_kwargs["user_agent"] = http_user_agent()
 
         if os.path.isfile(os.path.join(path, CONFIG_NAME)):
             config_file = os.path.join(path, CONFIG_NAME)

src/peft/utils/save_and_load.py

@@ -22,6 +22,7 @@
 from huggingface_hub import file_exists, hf_hub_download
 from huggingface_hub.errors import EntryNotFoundError, LocalEntryNotFoundError
 from safetensors.torch import load_file as safe_load_file
+from transformers.utils import http_user_agent
 
 from peft.mapping import PEFT_TYPE_TO_PREFIX_MAPPING
 
@@ -510,6 +511,9 @@ def get_hub_filename(use_safetensors=True):
             else weights_name
         )
 
+    if "user_agent" not in hf_hub_download_kwargs:
+        hf_hub_download_kwargs["user_agent"] = http_user_agent()
+
     if os.path.exists(os.path.join(path, SAFETENSORS_WEIGHTS_NAME)):
         filename = os.path.join(path, SAFETENSORS_WEIGHTS_NAME)
         use_safetensors = True