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

flake--

Browse source
This commit is contained in:
Sigbjørn Skjæret 2024-04-20 08:48:39 +02:00 • committed by GitHub
parent 8d36967a2c
commit c4e6f6f4b9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
gguf-py/scripts/gguf-new-metadata.py
View file

@ -25,6 +25,7 @@ class MetadataDetails:
value: Any value: Any
description: str = '' description: str = ''
def get_byteorder(reader: gguf.GGUFReader) -> gguf.GGUFEndian: def get_byteorder(reader: gguf.GGUFReader) -> gguf.GGUFEndian:
if np.uint32(1) == np.uint32(1).newbyteorder("<"): if np.uint32(1) == np.uint32(1).newbyteorder("<"):
# Host is little endian # Host is little endian
@ -67,12 +68,12 @@ def get_field_data(reader: gguf.GGUFReader, key: str) -> Any:
def find_token(token_list: Sequence[int], token: str) -> Sequence[int]: def find_token(token_list: Sequence[int], token: str) -> Sequence[int]:
token_ids = [index for index, value in enumerate(token_list) if value == token] token_ids = [index for index, value in enumerate(token_list) if value == token]
if len(token_ids) == 0: if len(token_ids) == 0:
raise LookupError(f'Unable to find "{token}" in token list!') raise LookupError(f'Unable to find "{token}" in token list!')
return token_ids return token_ids
def copy_with_new_metadata(reader: gguf.GGUFReader, writer: gguf.GGUFWriter, new_metadata: Mapping[str, MetadataDetails], remove_metadata: Sequence[str]) -> None: def copy_with_new_metadata(reader: gguf.GGUFReader, writer: gguf.GGUFWriter, new_metadata: Mapping[str, MetadataDetails], remove_metadata: Sequence[str]) -> None: