def from_pretrained()

in src/speechbox/restore.py [0:0]

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

    def from_pretrained(cls, pretrained_model_or_path):
        model = WhisperForConditionalGeneration.from_pretrained(pretrained_model_or_path, low_cpu_mem_usage=True)
        processor = WhisperProcessor.from_pretrained(pretrained_model_or_path)
        return cls(model, processor)