starling.models.transformer.DiTBlock
====================================

.. currentmodule:: starling.models.transformer

.. autoclass:: DiTBlock
   :members:
   :show-inheritance:
   :special-members: __init__, __call__

   
   
   .. rubric:: Methods

   .. autosummary::
      :nosignatures:
   
      ~DiTBlock.__init__
      ~DiTBlock.add_module
      ~DiTBlock.apply
      ~DiTBlock.bfloat16
      ~DiTBlock.buffers
      ~DiTBlock.children
      ~DiTBlock.compile
      ~DiTBlock.cpu
      ~DiTBlock.cuda
      ~DiTBlock.double
      ~DiTBlock.eval
      ~DiTBlock.extra_repr
      ~DiTBlock.float
      ~DiTBlock.forward
      ~DiTBlock.get_buffer
      ~DiTBlock.get_extra_state
      ~DiTBlock.get_parameter
      ~DiTBlock.get_submodule
      ~DiTBlock.half
      ~DiTBlock.ipu
      ~DiTBlock.load_state_dict
      ~DiTBlock.modules
      ~DiTBlock.mtia
      ~DiTBlock.named_buffers
      ~DiTBlock.named_children
      ~DiTBlock.named_modules
      ~DiTBlock.named_parameters
      ~DiTBlock.parameters
      ~DiTBlock.register_backward_hook
      ~DiTBlock.register_buffer
      ~DiTBlock.register_forward_hook
      ~DiTBlock.register_forward_pre_hook
      ~DiTBlock.register_full_backward_hook
      ~DiTBlock.register_full_backward_pre_hook
      ~DiTBlock.register_load_state_dict_post_hook
      ~DiTBlock.register_load_state_dict_pre_hook
      ~DiTBlock.register_module
      ~DiTBlock.register_parameter
      ~DiTBlock.register_state_dict_post_hook
      ~DiTBlock.register_state_dict_pre_hook
      ~DiTBlock.requires_grad_
      ~DiTBlock.set_extra_state
      ~DiTBlock.set_submodule
      ~DiTBlock.share_memory
      ~DiTBlock.state_dict
      ~DiTBlock.to
      ~DiTBlock.to_empty
      ~DiTBlock.train
      ~DiTBlock.type
      ~DiTBlock.xpu
      ~DiTBlock.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~DiTBlock.T_destination
      ~DiTBlock.call_super_init
      ~DiTBlock.dump_patches
      ~DiTBlock.training