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

BertSafeTensorsNameResolver

class BertSafeTensorsNameResolver : WeightNameResolver(source)

Resolves network DSL module paths to HuggingFace BERT tensor names.

Module tree paths → HuggingFace BERT naming: "MLP/embeddings/word_embeddings" + "word_embeddings.weight" → "bert.embeddings.word_embeddings.weight" "MLP/embeddings/LayerNorm" + "LayerNorm.weight" → "bert.embeddings.LayerNorm.weight" "MLP/encoder.layer.0/attention" + "attention.q_proj.weight" → "bert.encoder.layer.0.attention.self.query.weight" "MLP/encoder.layer.0/intermediate" + "intermediate.weight" → "bert.encoder.layer.0.intermediate.dense.weight" "MLP/encoder.layer.0/attn_ln" + "attn_ln.weight" → "bert.encoder.layer.0.attention.output.LayerNorm.weight" "MLP/encoder.layer.0/output_ln" + "output_ln.weight" → "bert.encoder.layer.0.output.LayerNorm.weight"

Constructors

constructor()

Functions

resolve

open override fun resolve(modulePath: String, paramName: String): String?

Resolve a module parameter path to the tensor name in the model file.