brainscore\_language.model\_helpers.modeling\_suma
==================================================

.. automodule:: brainscore_language.model_helpers.modeling_suma

   
   .. rubric:: Functions

   .. autosummary::
   
      custom_init_weights
      repeat_kv
   
   .. rubric:: Classes

   .. autosummary::
   
      BaseModelOutputWithPast
      Cache
      DynamicCache
      LlamaAttention
      LlamaPreTrainedModel
      LlamaRMSNorm
      SUMAConfig
      SUMADecoderLayer
      SUMAForCausalLM
      SUMAModel