From e856bfed3bbd18232c121b7f6a65ec3355cf2fd3 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Thu, 15 Feb 2024 15:05:15 +0200
Subject: [PATCH] hf : add support for --repo and --file

---
 scripts/hf.sh | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/scripts/hf.sh b/scripts/hf.sh
index bcd4426b4..1e9e5a6ea 100755
--- a/scripts/hf.sh
+++ b/scripts/hf.sh
@@ -5,6 +5,7 @@
 # Usage:
 #   ./main -m $(./examples/hf.sh https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/resolve/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
 #   ./main -m $(./examples/hf.sh --url https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/blob/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
+#   ./main -m $(./examples/hf.sh --repo TheBloke/Mixtral-8x7B-v0.1-GGUF --file mixtral-8x7b-v0.1.Q4_K_M.gguf)
 #
 
 # all logs go to stderr
@@ -13,7 +14,7 @@ function log {
 }
 
 function usage {
-    log "Usage: $0 --url <url> [--help]"
+    log "Usage: $0 [[--url] <url>] [--repo <repo>] [--file <file>] [-h|--help]"
     exit 1
 }
 
@@ -34,6 +35,8 @@ else
 fi
 
 url=""
+repo=""
+file=""
 
 # parse args
 while [[ $# -gt 0 ]]; do
@@ -42,6 +45,14 @@ while [[ $# -gt 0 ]]; do
             url="$2"
             shift 2
             ;;
+        --repo)
+            repo="$2"
+            shift 2
+            ;;
+        --file)
+            file="$2"
+            shift 2
+            ;;
         -h|--help)
             usage
             ;;
@@ -52,6 +63,10 @@ while [[ $# -gt 0 ]]; do
     esac
 done
 
+if [ -n "$repo" ] && [ -n "$file" ]; then
+    url="https://huggingface.co/$repo/resolve/main/$file"
+fi
+
 if [ -z "$url" ]; then
     log "[E] missing --url"
     usage