Skip to content

Commit 6c8a629

Browse files
authored
fix: Change BOS/EOS token ids for non-multilingual Whisper in model configs (#170)
## Description This PR fixes a change which was made by accident when adding code for multilingual Whisper. The aforementioned PR changed BOS and EOS for non-multilingual Whisper tiny, which should not be changed. ### Type of change - [x] Bug fix (non-breaking change which fixes an issue) - [ ] New feature (non-breaking change which adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected) - [ ] Documentation update (improves or adds clarity to existing documentation) ### Tested on - [ ] iOS - [ ] Android ### Testing instructions <!-- Provide step-by-step instructions on how to test your changes. Include setup details if necessary. --> ### Screenshots <!-- Add screenshots here, if applicable --> ### Related issues <!-- Link related issues here using #issue-number --> ### Checklist - [ ] I have performed a self-review of my code - [ ] I have commented my code, particularly in hard-to-understand areas - [ ] I have updated the documentation accordingly - [ ] My changes generate no new warnings ### Additional notes <!-- Include any additional information, assumptions, or context that reviewers might need to understand this PR. -->
1 parent a8e7f96 commit 6c8a629

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/constants/sttDefaults.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@ const whisperTinyModelConfig = {
2222
},
2323
tokenizer: {
2424
source: WHISPER_TOKENIZER,
25-
bos: 50258, // FIXME: this is a placeholder and needs to be changed
26-
eos: 50257, // FIXME: this is a placeholder and needs to be changed
25+
bos: 50257, // FIXME: this is a placeholder and needs to be changed
26+
eos: 50256, // FIXME: this is a placeholder and needs to be changed
2727
},
2828
isMultilingual: false,
2929
};

0 commit comments

Comments
 (0)