Skip to content

Commit 2d2bc99

Browse files
committed
convert-hf-to-gguf.py: add additional logging
1 parent 58d5a5d commit 2d2bc99

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

convert-hf-to-gguf.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2857,8 +2857,7 @@ def write_tensors(self):
28572857
# map tensor names
28582858
new_name = tensor_map.get_name(name, try_suffixes=(".weight", ".bias"))
28592859
if new_name is None:
2860-
print(f"Can not map tensor {name!r}")
2861-
sys.exit()
2860+
raise ValueError(f"Can not map tensor {name!r}")
28622861

28632862
n_dims = len(data.shape)
28642863
data_dtype = data.dtype
@@ -2875,7 +2874,7 @@ def write_tensors(self):
28752874
if self.ftype == 1 and data_dtype == np.float32 and n_dims == 2:
28762875
data = data.astype(np.float16)
28772876

2878-
print(f"{new_name}, n_dims = {n_dims}, {old_dtype} --> {data.dtype}")
2877+
logger.info(f"{new_name}, n_dims = {n_dims}, {old_dtype} --> {data.dtype}")
28792878

28802879
self.gguf_writer.add_tensor(new_name, data)
28812880

0 commit comments

Comments
 (0)