neural\_de.transformations.diffusion.unet package
=================================================

Submodules
----------

neural\_de.transformations.diffusion.unet.attention\_block module
-----------------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.attention_block
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.downsample module
-----------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.downsample
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.nn module
---------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.nn
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.qkv\_attention module
---------------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.qkv_attention
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.qkv\_attention\_legacy module
-----------------------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.qkv_attention_legacy
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.res\_block module
-----------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.res_block
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.timestep\_block module
----------------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.timestep_block
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.timestep\_embed\_sequential module
----------------------------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.timestep_embed_sequential
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.unet\_model module
------------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.unet_model
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.upsample module
---------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.upsample
   :members:
   :undoc-members:
   :show-inheritance:

neural\_de.transformations.diffusion.unet.utils module
------------------------------------------------------

.. automodule:: neural_de.transformations.diffusion.unet.utils
   :members:
   :undoc-members:
   :show-inheritance:

Module contents
---------------

.. automodule:: neural_de.transformations.diffusion.unet
   :members:
   :undoc-members:
   :show-inheritance: