For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python function

role_weights

role_weights()

max.pipelines.lib.registration_runtime.role_weights(state_dict, config, role)

source

Returns the weights of one role, named as its module names them.

A role with a prefix in DeploymentConfig.weight_prefixes takes the weights under it, with the prefix removed; one without takes them all.

Parameters:

• state_dict (Mapping[str, _WeightT])
• config (object)
• role (str)

Return type:

dict[str, _WeightT]