lm_polygraph.stat_calculators.greedy_visual_probs module

class lm_polygraph.stat_calculators.greedy_visual_probs.GreedyProbsVisualCalculator(output_attentions: bool = True, output_hidden_states: bool = True, n_alternatives: int = 10, samples_n: int = 10)[source]

Bases: StatCalculator

For WhiteboxModelvLLM model, at input texts batch calculates: * sampled texts * tokens of the sampled texts * probabilities of the sampled tokens generation

static meta_info() → Tuple[List[str], List[str]][source]: Returns the statistics and dependencies for the calculator.