vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gguf_dump.py: handle edge case about backticks on start or end of a string

Browse source
This commit is contained in:
brian khuu 2024-07-20 16:17:01 +10:00
parent 50d55d601f
commit 1949847a9c
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
gguf-py/scripts/gguf_dump.py
View file

@ -250,6 +250,11 @@ def dump_markdown_metadata(reader: GGUFReader, args: argparse.Namespace) -> None
# wrap string with appropriate number of backticks required to escape it # wrap string with appropriate number of backticks required to escape it
max_backticks = max((len(match.group(0)) for match in re.finditer(r'`+', value_string)), default=0) max_backticks = max((len(match.group(0)) for match in re.finditer(r'`+', value_string)), default=0)
inline_code_marker = '`' * (max_backticks + 1) inline_code_marker = '`' * (max_backticks + 1)
# If the string starts or ends with a backtick, add a space at the beginning and end
if value_string.startswith('`') or value_string.endswith('`'):
value_string = f" {value_string} "
return f"{inline_code_marker}{value_string}{inline_code_marker}" return f"{inline_code_marker}{value_string}{inline_code_marker}"
total_elements = len(field.data) total_elements = len(field.data)