Method
match_model_tensor_name
(self, name: str, key: gguf.MODEL_TENSOR, bid: int | None, suffix: str = ".weight")
Source from the content-addressed store, hash-verified
| 167 | yield name, data |
| 168 | |
| 169 | def match_model_tensor_name(self, name: str, key: gguf.MODEL_TENSOR, bid: int | None, suffix: str = ".weight") -> bool: |
| 170 | if key not in gguf.MODEL_TENSORS[self.model_arch]: |
| 171 | return False |
| 172 | key_name: str = gguf.TENSOR_NAMES[key] |
| 173 | if "{bid}" in key_name: |
| 174 | if bid is None: |
| 175 | return False |
| 176 | key_name = key_name.format(bid=bid) |
| 177 | else: |
| 178 | if bid is not None: |
| 179 | return False |
| 180 | return name == (key_name + suffix) |
| 181 | |
| 182 | def map_tensor_name(self, name: str, try_suffixes: Sequence[str] = (".weight", ".bias")) -> str: |
| 183 | new_name = self.tensor_map.get_name(key=name, try_suffixes=try_suffixes) |
Tested by
no test coverage detected