Upload folder using huggingface_hub

README.md

@@ -9,5 +9,42 @@ tags:
 # Yi-6B-Chat-MNN
 
 ## Introduction
+This model is a 4-bit quantized version of the MNN model exported from [Yi-6B-Chat](https://modelscope.cn/models/01ai/Yi-6B-Chat/summary) using [llmexport](https://github.com/alibaba/MNN/tree/master/transformers/llm/export).
 
-This model is a 4-bit quantized version of the MNN model exported from Yi-6B-Chat using [llm-export](https://github.com/wangzhaode/llm-export).
+## Download
+```bash
+# install huggingface
+pip install huggingface
+```
+```bash
+# shell download
+huggingface download --model 'taobao-mnn/Yi-6B-Chat-MNN' --local_dir 'path/to/dir'
+```
+```python
+# SDK download
+from huggingface_hub import snapshot_download
+model_dir = snapshot_download('taobao-mnn/Yi-6B-Chat-MNN')
+```
+
+```bash
+# git clone
+git clone https://www.modelscope.cn/taobao-mnn/Yi-6B-Chat-MNN
+```
+
+## Usage
+```bash
+# clone MNN source
+git clone https://github.com/alibaba/MNN.git
+
+# compile
+cd MNN
+mkdir build && cd build
+cmake .. -DMNN_LOW_MEMORY=true -DMNN_CPU_WEIGHT_DEQUANT_GEMM=true -DMNN_BUILD_LLM=true -DMNN_SUPPORT_TRANSFORMER_FUSE=true
+make -j
+
+# run
+./llm_demo /path/to/Yi-6B-Chat-MNN/config.json prompt.txt
+```
+
+## Document
+[MNN-LLM](https://mnn-docs.readthedocs.io/en/latest/transformers/llm.html#)

config.json

@@ -1,9 +1,8 @@
 {
-    "llm_model": "yi-6b-int4.mnn",
-    "llm_weight": "yi-6b-int4.mnn.weight",
-
+    "llm_model": "llm.mnn",
+    "llm_weight": "llm.mnn.weight",
     "backend_type": "cpu",
     "thread_num": 4,
     "precision": "low",
     "memory": "low"
-}
+}

llm.mnn

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b0c4e9b4b6d0a7c80c54de96b57afa9ba09e7e3b06e483bd33352d13bcbc342b
+size 2637408

llm.mnn.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50ade3ec36f8a4454258acf64673863a960a2562a95ba7e79aa886efe00b4fc9
+size 16701691

llm.mnn.weight

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57245541501bf1f06f5c7c86f35d5416dfae0fe1d8a8169da0748e8b15e95ba5
+size 3261731670

llm_config.json

@@ -10,6 +10,5 @@
         128
     ],
     "prompt_template": "<|im_start|> user\n%s<|im_end|>\n<|im_start|> assistant\n",
-    "is_visual": false,
-    "is_single": true
+    "is_visual": false
 }