Skip to content

Commit 8fd6271

Browse files
committed
fix(convert): Remove unnecessary safe dict get
#1250 Branch: BiasTensors-1250 Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
1 parent 0ddaf93 commit 8fd6271

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchchat/cli/convert_hf_checkpoint.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,9 @@ def load_safetensors():
142142
if "layers" in key:
143143
abstract_key = re.sub(r"(\d+)", "{}", key)
144144
layer_num = re.search(r"\d+", key).group(0)
145-
new_key = weight_map.get(abstract_key, abstract_key)
145+
new_key = weight_map[abstract_key]
146+
if new_key is None:
147+
continue
146148
new_key = new_key.format(layer_num)
147149
else:
148150
new_key = weight_map[key]

0 commit comments

Comments
 (0)