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

type consistency in format_n_bytes_to_str

Browse source
This commit is contained in:
Christian Zhou-Zheng 2024-06-06 08:43:26 -04:00
parent 3328b0a991
commit 1cbab22225
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
gguf-py/gguf/gguf_manager.py
View file

@ -255,9 +255,9 @@ class GGUFManager(GGUFWriter):
def format_n_bytes_to_str(num: int) -> str: def format_n_bytes_to_str(num: int) -> str:
if num == METADATA_ONLY_INDICATOR: if num == METADATA_ONLY_INDICATOR:
return "negligible - metadata only" return "negligible - metadata only"
num = float(num) fnum = float(num)
for unit in ("", "K", "M", "G"): for unit in ("", "K", "M", "G"):
if abs(num) < 1024.0: if abs(fnum) < 1024.0:
return f"{num:3.1f}{unit}" return f"{fnum:3.1f}{unit}"
num /= 1024.0 fnum /= 1024.0
return f"{num:.1f}T - over 1TB, --split recommended" return f"{fnum:.1f}T - over 1TB, --split recommended"