From df175f7c60210cc900b28408f4196c7813d5fdfd Mon Sep 17 00:00:00 2001
From: Sourabrata Bose <bose.sourabrata21century@gmail.com>
Date: Wed, 17 Apr 2024 15:34:52 +0530
Subject: [PATCH] Added --hf-token argument support

---
 common/common.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/common/common.cpp b/common/common.cpp
index cf69535e2..5e9f7fa05 100644
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -767,6 +767,15 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
         params.hf_file = argv[i];
         return true;
     }
+    if (arg == "-hft" || arg == "--hf-token") {
+        params.hf_token = NULL;
+        if (++i >= argc) {
+            invalid_param = true;
+            return true;
+        }
+        params.hf_token = argv[i];
+        return true;
+    }
     if (arg == "--lora") {
         if (++i >= argc) {
             invalid_param = true;
@@ -2089,6 +2098,12 @@ struct llama_model * llama_load_model_from_url(
         return NULL;
     }
 
+    if(params.hf_token!=NULL){
+        struct curl_slist *headers = NULL;
+        headers = curl_slist_append(headers, "Authorization: Bearer " + params.hf_token);
+        curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
+    }
+
     if (!llama_download_file(curl, model_url, path_model)) {
         return NULL;
     }