transolver_config
=================

.. py:module:: transolver_config


Classes
-------

.. autoapisummary::

   transolver_config.TransolverConfig
   transolver_config.TransolverPlusPlusConfig


Module Contents
---------------

.. py:class:: TransolverConfig(/, **data)

   Bases: :py:obj:`emmi.schemas.models.architectures.transformer_config.TransformerConfig`


   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.


   .. py:attribute:: attention_constructor
      :type:  type

      Constructor of the attention module. Defaults to DotProductAttention..


   .. py:attribute:: attention_arguments
      :type:  dict

      Additional arguments for the attention module that are only needed for a specific attention impelentation.


.. py:class:: TransolverPlusPlusConfig(/, **data)

   Bases: :py:obj:`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.


   .. py:attribute:: attention_constructor
      :type:  type

      Constructor of the attention module. Defaults to DotProductAttention..


   .. py:attribute:: attention_arguments
      :type:  dict

      Additional arguments for the attention module that are only needed for a specific attention impelentation.