def freeze_encoders()

in parler_tts/modeling_parler_tts.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def freeze_encoders(self, freeze_text_encoder=True):
        if freeze_text_encoder:
            for param in self.text_encoder.parameters():
                param.requires_grad = False
            self.text_encoder._requires_grad = False

        for param in self.audio_encoder.parameters():
            param.requires_grad = False
        self.audio_encoder._requires_grad = False