src/transformers/models/unispeech/modeling_unispeech.py/UniSpeechForPreTraining/set_gumbel_temperature
class UniSpeechForPreTraining:    def set_gumbel_temperature(self, temperature: int):
        """
        Set the Gumbel softmax temperature to a given value. Only necessary for training
        """
        self.quantizer.temperature = temperature