nemo_rl.models.generation

Sous-packages

vllm

Sous-modules

interfaces
fp8

Contenu du module

Fonctions

Nom	Description
`configure_generation_config`	Appliquer des configurations spécifiques à la configuration de génération.

Données

TokenizerType

API

nemo_rl.models.generation.TokenizerType

Valeur: 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

Appliquer des configurations spécifiques à la configuration de génération.