Skip to content

Commit 0ac626a

Browse files
authored
fix update vocab param loading (#184)
1 parent 7947a6b commit 0ac626a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eole/models/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,7 @@ def load_safe_state_dict(
628628
ckpt_t = updated_params[name + "." + param_name]
629629
self._load_param(name, module, param_name, param, buf_list, ckpt_t, offset)
630630
keyfound[name + "." + param_name] = True
631-
if name + "." + param_name in keys_shard.keys():
631+
elif name + "." + param_name in keys_shard.keys():
632632

633633
ckpt_t = f[keys_shard[name + "." + param_name]].get_tensor(name + "." + param_name)
634634
self._load_param(name, module, param_name, param, buf_list, ckpt_t, offset)

0 commit comments

Comments
 (0)