megatron.text\_generation.forward\_step
=======================================

.. rubric:: Description

.. automodule:: megatron.text_generation.forward_step

.. currentmodule:: megatron.text_generation.forward_step


.. rubric:: Classes

.. autosummary::
    :toctree: .
    
    ForwardStep
    
    InferenceParams