nemo_rl.models.generation#

Submodules#

Package Contents#

Functions#

configure_generation_config

Apply specific configurations to generation config.

Data#

TokenizerType

API#

nemo_rl.models.generation.TokenizerType#

None

nemo_rl.models.generation.configure_generation_config(
config: nemo_rl.models.generation.interfaces.GenerationConfig,
tokenizer: nemo_rl.models.generation.TokenizerType,
is_eval=False,
) → nemo_rl.models.generation.interfaces.GenerationConfig[source]#

Apply specific configurations to generation config.