lm_polygraph.utils.ensemble_utils.ensemble_generator module

class lm_polygraph.utils.ensemble_utils.ensemble_generator.EnsembleGenerationMixin[source]

Bases: EnsembleBeamSearchMixin, EnsembleSampleMixin, EnsembleGreedyMixin, GenerationMixin

add_ensemble_models(models, devices)[source]
property base_seed
calculate_entropy_based_measures(enable=True)[source]
property ensembling_mode
property mc
property mc_models_num
property mc_seeds
property models
property models_beam_logits_iter
property tokenizer