skainet-io-safetensors/sk.ainet.io.safetensors/StreamingShardedSafeTensorsReader/loadTensorData

loadTensorData

fun loadTensorData(name: String): ByteArray(source)

Load tensor data by name.

Return

Raw bytes for the tensor

Parameters

name

The tensor name

Throws

IllegalArgumentException

if tensor not found

IllegalStateException

if the containing shard was not loaded

fun loadTensorData(tensor: ShardedTensorInfo): ByteArray(source)

Load tensor data for a specific tensor.

Return

Raw bytes for the tensor

Parameters

tensor

The tensor info from tensors list

Throws

IllegalStateException

if the containing shard was not loaded