Skip to main content

load_weights

Function load_weights 

Source
pub fn load_weights(
    ctx: &Arc<WgpuContext>,
    weights_ctx: &WeightsContext,
    tensors: &HashMap<String, Tensor>,
) -> Result<(), LoaderError>
Expand description

Overwrite each tensor in tensors with the weight of the same name in weights_ctx; names that are missing are logged and skipped.