(self, n_part: int)
| 615 | return UnquantizedTensor(permute(self.ndarray[r * n_part : r * n_part + r, ...], n_head, n_head_kv)) |
| 616 | |
| 617 | def part(self, n_part: int) -> UnquantizedTensor: |
| 618 | r = self.ndarray.shape[0] // 3 |
| 619 | return UnquantizedTensor(self.ndarray[r * n_part : r * n_part + r, ...]) |
| 620 | |
| 621 | def permute(self, n_head: int, n_head_kv: int) -> UnquantizedTensor: |
| 622 | return UnquantizedTensor(permute(self.ndarray, n_head, n_head_kv)) |
nothing calls this directly
no test coverage detected