add windows download script.

wangzhaode · Sep 18, 2023 · 43bd439 · 43bd439
1 parent c858ab2
commit 43bd439
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -40,7 +40,11 @@
 ### 下载int4模型
 ```
 # <model> like `chatglm-6b`
+# linux/macos
 ./script/download_model.sh <model>
+
+# windows
+./script/download_model.ps1 <model>
 ```
 
 ## 构建
@@ -102,4 +106,4 @@
 - [cpp-httplib](https://github.com/yhirose/cpp-httplib)
 - [chatgpt-web](https://github.com/xqdoo00o/chatgpt-web)
 - [cppjieba](https://github.com/yanyiwu/cppjieba)
-- [ChatViewDemo](https://github.com/BrettFX/ChatViewDemo)
+- [ChatViewDemo](https://github.com/BrettFX/ChatViewDemo)
diff --git a/script/download_model.ps1 b/script/download_model.ps1
@@ -0,0 +1,12 @@
+param(
+     $model
+)
+mkdir $model
+cd $model
+wget -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/embedding.mnn -OutFile embedding.mnn
+wget -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/lm.mnn -OutFile lm.mnn
+for($i=1; $i -lt 27; $i=$i+1)
+{   
+    wget -Uri https://github.com/wangzhaode/mnn-llm/releases/download/$model-mnn/block_$i.mnn  -OutFile  block_$i.mnn
+}
+cd ..