Skip to content

Temperatures - support for list of values #520

Description

@Bykiev

Is your feature request related to a problem? Please describe.
Right now it's not possible to pass multiple temperature values

Describe the solution you'd like
https://huggingface.co/docs/transformers/en/model_doc/whisper#transformers.WhisperForConditionalGeneration.generate.temperature

temperature (float or list of float, optional) — The temperature to be used for generation. Passing a single float value and do_sample=True activates generation using sampling. For long-form transcription, temperature fallback can be activated by passing a list of float values such as (0.0, 0.2, 0.4, 0.6, 0.8, 1.0). As shown in the the Whisper paper, this can help to improve performance.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions