Skip to main content

Python class

ArchConfig

ArchConfigโ€‹

class max.pipelines.lib.interfaces.ArchConfig(*args, **kwargs)

source

Bases: Protocol

Config for a model architecture.

get_max_seq_len()โ€‹

get_max_seq_len()

source

Returns the default maximum sequence length for the model.

Subclasses should determine whether this value can be overridden by setting the --max-length (pipeline_config.model.max_length) flag.

Return type:

int

initialize()โ€‹

classmethod initialize(pipeline_config)

source

Initialize the config from a PipelineConfig.

Parameters:

pipeline_config (PipelineConfig)

Return type:

Self