resolve

abstract fun resolve(modulePath: String, paramName: String): String?(source)

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

Return

the tensor name in the model file, or null if no mapping exists

Parameters

modulePath

the path from the module tree (e.g. "MLP/blk.0/attn")

paramName

the parameter name (e.g. "attn.q_proj.weight")