MCPcopy Create free account
hub / github.com/Pints-AI/1.5-Pints / load

Function load

tokenizer/convert/convert.py:706–709  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

704 concatenated_shape[axis] = sum(tensor.shape[axis] for tensor in lazy_tensors)
705
706 def load() -> UnquantizedTensor:
707 ndarrays = [load_unquantized(tensor) for tensor in lazy_tensors]
708 concatenated: NDArray = np.concatenate(ndarrays, axis=axis)
709 return UnquantizedTensor(concatenated)
710 description = 'concatenated[[' + '] | ['.join(lt.description for lt in lazy_tensors) + ']]'
711 return LazyTensor(load, concatenated_shape, lazy_tensors[0].data_type, description)
712 return {name: convert(name) for name in names}

Callers 5

load_datasetMethod · 0.85
load_datasetMethod · 0.85
load_datasetMethod · 0.85
load_datasetMethod · 0.85
load_datasetMethod · 0.85

Calls 6

load_unquantizedFunction · 0.85
UnquantizedTensorClass · 0.85
permuteMethod · 0.45
loadMethod · 0.45
permute_partMethod · 0.45
partMethod · 0.45

Tested by

no test coverage detected