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

Make examples executable, formatting changes

Browse source
This commit is contained in:
KerfuffleV2 2023-11-09 00:25:20 -07:00
parent 855486c912
commit 2360aaadb4
3 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
gguf-py/examples/dump_gguf.py Normal file → Executable file
View file

@ -7,6 +7,7 @@ sys.path.insert(0, str(Path(__file__).parent.parent))
from gguf import GGUFReader, GGUFValueType # noqa: E402 from gguf import GGUFReader, GGUFValueType # noqa: E402
def dump_gguf(filename: str) -> None: def dump_gguf(filename: str) -> None:
print(f'* Loading: {filename}') print(f'* Loading: {filename}')
reader = GGUFReader(filename, 'r') reader = GGUFReader(filename, 'r')
@ -34,6 +35,7 @@ def dump_gguf(filename: str) -> None:
prettydims = ', '.join('{0:5}'.format(d) for d in list(tensor.shape) + [1] * (4 - len(tensor.shape))) prettydims = ', '.join('{0:5}'.format(d) for d in list(tensor.shape) + [1] * (4 - len(tensor.shape)))
print(f' {n:5}: {tensor.n_elements:10} | {prettydims} | {tensor.tensor_type.name:7} | {tensor.name}') print(f' {n:5}: {tensor.n_elements:10} | {prettydims} | {tensor.tensor_type.name:7} | {tensor.name}')
if __name__ == '__main__': if __name__ == '__main__':
if len(sys.argv) < 2: if len(sys.argv) < 2:
print('dump_gguf: Error: Specify an input file', file = sys.stderr) print('dump_gguf: Error: Specify an input file', file = sys.stderr)

2
gguf-py/examples/modify_gguf.py Normal file → Executable file
View file

@ -7,6 +7,7 @@ sys.path.insert(0, str(Path(__file__).parent.parent))
from gguf import GGUFReader # noqa: E402 from gguf import GGUFReader # noqa: E402
def change_gguf(reader: GGUFReader, key: str, value: str) -> None: def change_gguf(reader: GGUFReader, key: str, value: str) -> None:
field = reader.get_field(key) field = reader.get_field(key)
if field is None: if field is None:
@ -32,6 +33,7 @@ def change_gguf(reader: GGUFReader, key: str, value: str) -> None:
field.parts[field.data[0]][0] = new_value field.parts[field.data[0]][0] = new_value
print('* Field changed. Successful completion.') print('* Field changed. Successful completion.')
if __name__ == '__main__': if __name__ == '__main__':
if len(sys.argv) < 4: if len(sys.argv) < 4:
print( print(

5
gguf-py/examples/writer.py Normal file → Executable file
View file

@ -9,6 +9,7 @@ sys.path.insert(0, str(Path(__file__).parent.parent))
from gguf import GGUFWriter # noqa: E402 from gguf import GGUFWriter # noqa: E402
# Example usage: # Example usage:
def writer_example() -> None: def writer_example() -> None:
# Example usage with a file # Example usage with a file
@ -34,4 +35,6 @@ def writer_example() -> None:
gguf_writer.close() gguf_writer.close()
writer_example()
if __name__ == '__main__':
writer_example()