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