| import torch | |
| from transformers import WhisperPreTrainedModel, WhisperConfig | |
| from transformers.models.whisper.modeling_whisper import WhisperEncoder | |
| class WhisperEncoderOnlyModel(WhisperPreTrainedModel): | |
| config_class = WhisperConfig | |
| def __init__(self, config): | |
| super().__init__(config) | |
| self.encoder = WhisperEncoder(config) | |
| self.post_init() | |
| def forward(self, input_features, **kwargs): | |
| return self.encoder(input_features, **kwargs) |