Update model.py
Browse files
model.py
CHANGED
@@ -6,10 +6,9 @@ from typing import Iterator
|
|
6 |
model_id = 'theohlong/baichuan2_13b-GGML'
|
7 |
|
8 |
from huggingface_hub import snapshot_download,hf_hub_download
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
hf_hub_download(repo_id="baichuan-inc/Baichuan-13B-Chat",local_dir="./", filename="tokenizer.model")
|
13 |
from llama_cpp import Llama
|
14 |
llm = Llama(model_path="./ggml-model-q4_1.bin", n_ctx=4096,seed=-1)
|
15 |
|
|
|
6 |
model_id = 'theohlong/baichuan2_13b-GGML'
|
7 |
|
8 |
from huggingface_hub import snapshot_download,hf_hub_download
|
9 |
+
|
10 |
+
snapshot_download(model_id, local_dir="./")
|
11 |
+
hf_hub_download(repo_id="baichuan-inc/Baichuan-13B-Chat-4bits",local_dir="./", filename="tokenizer.model")
|
|
|
12 |
from llama_cpp import Llama
|
13 |
llm = Llama(model_path="./ggml-model-q4_1.bin", n_ctx=4096,seed=-1)
|
14 |
|