vit
===

.. automodule:: mfai.pytorch.models.vit
   :members:
   :undoc-members:
   :show-inheritance: