foo

Author: AngledLuffa

stanza/models/depparse/model.py

@@ -359,6 +359,7 @@ class EnsembleGraphParser(BaseParser):
     def __init__(self, args, vocab, models):
         super().__init__(args, vocab)
         self.models = nn.ModuleList(models)
+        self.fallback = self.models[0].fallback
 
     def get_params(self, skip_modules):
         params = []
@@ -406,7 +407,7 @@ def forward(self, word, word_mask, wordchars, wordchars_mask, upos, xpos, ufeats
     def predict(self, word, word_mask, wordchars, wordchars_mask, upos, xpos, ufeats, pretrained, lemma, head, deprel, word_orig_idx, sentlens, wordlens, text):
         batch_size = word.size(0)
         _, preds = self(word, word_mask, wordchars, wordchars_mask, upos, xpos, ufeats, pretrained, lemma, head, deprel, word_orig_idx, sentlens, wordlens, text)
-        pred_tokens = GraphParser.decode_graph_predictions(self.vocab['deprel'], batch_size, preds, sentlens)
+        pred_tokens = GraphParser.decode_graph_predictions(self.vocab['deprel'], self.fallback, batch_size, preds, sentlens)
         return pred_tokens
 
     def get_device(self):