Skip to content

Commit a6f54de

Browse files
committed
convert-hf-to-gguf.py: add additional logging
1 parent 81a9d17 commit a6f54de

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
@@ -2645,8 +2645,7 @@ def write_tensors(self):
26452645
# map tensor names
26462646
new_name = tensor_map.get_name(name, try_suffixes=(".weight", ".bias"))
26472647
if new_name is None:
2648-
print(f"Can not map tensor {name!r}")
2649-
sys.exit()
2648+
raise ValueError(f"Can not map tensor {name!r}")
26502649

26512650
n_dims = len(data.shape)
26522651
data_dtype = data.dtype
@@ -2663,7 +2662,7 @@ def write_tensors(self):
26632662
if self.ftype == 1 and data_dtype == np.float32 and n_dims == 2:
26642663
data = data.astype(np.float16)
26652664

2666-
print(f"{new_name}, n_dims = {n_dims}, {old_dtype} --> {data.dtype}")
2665+
logger.info(f"{new_name}, n_dims = {n_dims}, {old_dtype} --> {data.dtype}")
26672666

26682667
self.gguf_writer.add_tensor(new_name, data)
26692668

0 commit comments

Comments
 (0)