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
stage_layers
stage_layers()
max.pipelines.lib.pipeline_parallel.stage_layers(pipeline_config, num_layers)
Returns the decoder layers the config builds.
Every layer, unless the config carries a pipeline stage.
-
Parameters:
-
- pipeline_config (PipelineConfig) – The config a model builds from.
- num_layers (int) – The decoder’s layer count, checked against the stage.
-
Raises:
-
ValueError – If the stage’s layers do not fit the decoder.
-
Return type: