Cran-May commited on
Commit
48b63a9
Β·
1 Parent(s): fb3bb87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -45,10 +45,11 @@ def get_system_tokens(model):
45
  return get_message_tokens(model, **system_message)
46
 
47
 
48
- repo_name = "TheBloke/Nanbeige-16B-Chat-GGUF"
49
- model_name = "nanbeige-16b-chat.Q4_K_M.gguf"
50
 
51
  snapshot_download(repo_id=repo_name, local_dir=".", allow_patterns=model_name)
 
52
 
53
  model = Llama(
54
  model_path=model_name,
 
45
  return get_message_tokens(model, **system_message)
46
 
47
 
48
+ repo_name = "theohlong/baichuan2_13b-GGML"
49
+ model_name = "ggml-model-q4_1.bin"
50
 
51
  snapshot_download(repo_id=repo_name, local_dir=".", allow_patterns=model_name)
52
+ hf_hub_download(repo_id="baichuan-inc/Baichuan-13B-Chat",local_dir="./", filename="tokenizer.model")
53
 
54
  model = Llama(
55
  model_path=model_name,