treat index as authoritative
This commit is contained in:
parent
a458664fc8
commit
077f1bdd8c
|
|
@ -217,7 +217,7 @@ class ModelBase:
|
|||
if weight_map is None or not isinstance(weight_map, dict):
|
||||
raise ValueError(f"Can't load 'weight_map' from {index_name!r}")
|
||||
tensor_names_from_index.update(weight_map.keys())
|
||||
part_dict: dict[str, None] = dict.fromkeys(list(weight_map.values()) + part_names, None)
|
||||
part_dict: dict[str, None] = dict.fromkeys(weight_map.values(), None)
|
||||
part_names = list(part_dict.keys())
|
||||
else:
|
||||
weight_map = {}
|
||||
|
|
|
|||
Loading…
Reference in New Issue