From db65f2b4e663531f48c545117ac7348d8f32bb4c Mon Sep 17 00:00:00 2001
From: xieincz <39424496+xieincz@users.noreply.github.com>
Date: Thu, 25 Apr 2024 10:41:08 +0800
Subject: [PATCH] =?UTF-8?q?=E9=BB=98=E8=AE=A4=E4=BD=BF=E7=94=A8=E9=95=9C?=
 =?UTF-8?q?=E5=83=8F=E7=AB=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/huggingface-go/main.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/huggingface-go/main.go b/src/huggingface-go/main.go
index cc9253e..e01ed94 100644
--- a/src/huggingface-go/main.go
+++ b/src/huggingface-go/main.go
@@ -24,10 +24,14 @@ var huggingfaceHead string
 
 func main() {
 	var url, targetParentFolder, proxyURLHead, homepage string
+	var disableDefaultMirror bool
 	flag.StringVar(&url, "u", "", "huggingface url, such as: https://hf-mirror.com/Finnish-NLP/t5-large-nl36-finnish/tree/main")
 	flag.StringVar(&targetParentFolder, "f", "./", "path to your target folder")
 	flag.StringVar(&proxyURLHead, "p", "", "proxy url, leave it empty if you don't need it")
 	flag.StringVar(&homepage, "homepage", "https://github.com/xieincz/huggingface-go", "homepage url of this tool")
+	flag.StringVar(&huggingfaceHead, "m", "https://hf-mirror.com", "mirror url of huggingface, use this if you want to use a different mirror, use -d to disable default mirror")
+	flag.BoolVar(&disableDefaultMirror, "d", false, "disable default mirror")
+
 	flag.Parse()
 
 	if url == "" {
@@ -49,7 +53,10 @@ func main() {
 
 	//提取出域名
 	tmp = strings.Split(url, "/")
-	huggingfaceHead = tmp[0] + "//" + tmp[2] //e.g. https://huggingface.co
+	if disableDefaultMirror {
+		huggingfaceHead = tmp[0] + "//" + tmp[2] //e.g. https://huggingface.co
+		fmt.Printf("Mirror has been disabled, using %s as the mirror\n", huggingfaceHead)
+	}
 
 	fmt.Printf("Model/Datasets name: %s\n", modelName)
 	fmt.Printf("Model/Datasets url: %s\n", modelURL)