    def warm_up(self, sample, repeat):
        """Warm up the model"""
        for _i in range(repeat):
            self.forward(sample)
        logger.info(f"Model warmed up by running inference {repeat} times")
Warm up the model
examples/speech_to_speech/benchmarking/core.py