noether.core.schemas.models.transolver

Classes

TransolverConfig	Configuration for a Transolver model.
TransolverPlusPlusConfig	Configuration for a Transolver++ model.

Module Contents

class noether.core.schemas.models.transolver.TransolverConfig(/, **data)

Bases: noether.core.schemas.models.transformer.TransformerConfig, noether.core.schemas.models.base.ModelBaseConfig

Configuration for a Transolver model.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

model_config

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

attention_arguments: dict

set_attention_constructor()

Set attention_constructor in transformer_block_config based on data_specs.

Return type:

TransolverConfig

class noether.core.schemas.models.transolver.TransolverPlusPlusConfig(/, **data)

Bases: TransolverConfig

Configuration for a Transolver++ model.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

set_attention_constructor()

Set attention_constructor in transformer_block_config based on data_specs.

Return type:

TransolverPlusPlusConfig