Skip to content

Commit 3511437

Browse files
committed
Reenable missing experts error
1 parent 8d6ac42 commit 3511437

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2891,6 +2891,15 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
28912891
return []
28922892
return [(self.map_tensor_name(name), data_torch)]
28932893

2894+
def prepare_tensors(self):
2895+
super().prepare_tensors()
2896+
2897+
if self._experts is not None:
2898+
# flatten `list[dict[str, Tensor]]` into `list[str]`
2899+
experts = [k for d in self._experts for k in d.keys()]
2900+
if len(experts) > 0:
2901+
raise ValueError(f"Unprocessed experts: {experts}")
2902+
28942903

28952904
@ModelBase.register(
28962905
"Qwen2VLModel",

0 commit comments

Comments
 (0)