promptbind.models.model\_utils
==============================

.. automodule:: promptbind.models.model_utils

   
   .. rubric:: Functions

   .. autosummary::
   
      flatten_final_dims
      permute_final_dims
   
   .. rubric:: Classes

   .. autosummary::
   
      Attention
      GaussianSmearing
      InteractionModule
      RBFDistanceModule
      Transition