src/transformers/models/wav2vec2/modeling_wav2vec2.py/Wav2Vec2ForPreTraining/set_gumbel_temperature
class Wav2Vec2ForPreTraining:    def set_gumbel_temperature(self, temperature: int):
        """
        Set the Gumbel softmax temperature to a given value. Only necessary for training
        """
        return self.quantizer.set_temperature(temperature)