MCPcopy Create free account
hub / github.com/Tiiny-AI/PowerInfer / part

Method part

convert.py:496–498  ·  view source on GitHub ↗
(self, n_part: int)

Source from the content-addressed store, hash-verified

494 return UnquantizedTensor(permute(self.ndarray[r * n_part : r * n_part + r, ...], n_head, n_head_kv))
495
496 def part(self, n_part: int) -> UnquantizedTensor:
497 r = self.ndarray.shape[0] // 3
498 return UnquantizedTensor(self.ndarray[r * n_part : r * n_part + r, ...])
499
500 def permute(self, n_head: int, n_head_kv: int) -> UnquantizedTensor:
501 return UnquantizedTensor(permute(self.ndarray, n_head, n_head_kv))

Callers

nothing calls this directly

Calls 1

UnquantizedTensorClass · 0.70

Tested by

no test coverage detected