We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 67b28a5 commit 1f8e1c5Copy full SHA for 1f8e1c5
wenet/transformer/asr_model.py
@@ -50,8 +50,16 @@ def __init__(
50
51
super().__init__()
52
# note that eos is the same as sos (equivalent ID)
53
- self.sos = special_tokens.get("sos", vocab_size - 1)
54
- self.eos = special_tokens.get("eos", vocab_size - 1)
+ self.sos = (
+ vocab_size - 1
55
+ if special_tokens is None
56
+ else special_tokens.get("sos", vocab_size - 1)
57
+ )
58
+ self.eos = (
59
60
61
+ else special_tokens.get("eos", vocab_size - 1)
62
63
self.vocab_size = vocab_size
64
self.special_tokens = special_tokens
65
self.ignore_id = ignore_id
0 commit comments