Skip to content

Commit eff383b

Browse files
committed
invoking __call__ instead of forward()
1 parent 02aa851 commit eff383b

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,10 @@ We used Python 3.9.9 and [PyTorch](https://pytorch.org/) 1.10.1 to train and tes
2121

2222
pip install git+https://github.com/openai/whisper.git
2323

24+
To update the package to the latest version of this repository, please run:
25+
26+
pip install --upgrade --no-deps --force-reinstall git+https://github.com/openai/whisper.git
27+
2428
It also requires the command-line tool [`ffmpeg`](https://ffmpeg.org/) to be installed on your system, which is available from most package managers:
2529

2630
```bash

whisper/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -214,10 +214,10 @@ def __init__(self, dims: ModelDimensions):
214214
)
215215

216216
def embed_audio(self, mel: torch.Tensor):
217-
return self.encoder.forward(mel)
217+
return self.encoder(mel)
218218

219219
def logits(self, tokens: torch.Tensor, audio_features: torch.Tensor):
220-
return self.decoder.forward(tokens, audio_features)
220+
return self.decoder(tokens, audio_features)
221221

222222
def forward(self, mel: torch.Tensor, tokens: torch.Tensor) -> Dict[str, torch.Tensor]:
223223
return self.decoder(tokens, self.encoder(mel))

0 commit comments

Comments
 (0)