tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-06-26 19:55:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

llama : support InternLM2 (#5184)

Browse Source 
* support InternLM2 inference
  * add add_space_prefix KV pair
This commit is contained in:
Guoteng
2024-02-01 17:19:51 +08:00
committed by GitHub
parent 1cfb5372cf
commit ce32060198
5 changed files with 387 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gguf-py/gguf/gguf_writer.py
View File

@ -411,6 +411,9 @@ class GGUFWriter:
def add_add_eos_token(self, value: bool) -> None:
self.add_bool(Keys.Tokenizer.ADD_EOS, value)
def add_add_space_prefix(self, value: bool) -> None:
self.add_bool(Keys.Tokenizer.ADD_PREFIX, value)
def add_chat_template(self, value: str) -> None:
self.add_string(Keys.Tokenizer.CHAT_TEMPLATE, value)