Update dictionary.py revert back to old solution

lydianish · Jul 24, 2024 · e67ba8a · e67ba8a
1 parent 5fbe8b4
commit e67ba8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fairseq/data/dictionary.py b/fairseq/data/dictionary.py
@@ -131,7 +131,7 @@ def add_symbol(self, word, n=1, overwrite=True):
             self.count[idx] = self.count[idx] + n
             return idx
         else:
-            idx = len(self.indices)
+            idx = len(self.symbols)
             self.indices[word] = idx
             self.symbols.append(word)
             self.count.append(n)