skainet-io-core/sk.ainet.io.weights

Package-level declarations

Types

Resolves network DSL module paths to HuggingFace BERT tensor names.

Resolves network DSL module paths to GGUF tensor names for LLaMA-family models.

Resolves network DSL module paths to SafeTensors (HuggingFace) tensor names.

data class MappingConfig(val usePathBasedMatching: Boolean = true, val fallbackToShapeMatching: Boolean = true, val debug: Boolean = false, val nameResolver: WeightNameResolver? = null)

Configuration for weight mapping behavior.

data class WeightLoadResult<T : DType, V>(val tensors: List<WeightTensor<T, V>>, val skipped: List<String> = emptyList())

Result of loading weight tensors from a model file.

Unified weight mapping utility for applying loaded weights to SKaiNET modules.

Resolves module parameter paths (from the DSL module tree) to weight tensor names (from model files like GGUF or SafeTensors).

data class WeightTensor<T : DType, V>(val name: String, val shape: List<Int>, val tensor: Tensor<T, V>, val isBias: Boolean = name.lowercase().contains("bias"))

Format-agnostic representation of a weight tensor loaded from any model format (ONNX, GGUF, SafeTensors, etc.).