forgot the eval

BigEmperor26 · Dec 15, 2023 · 0246d68 · 0246d68
1 parent e015cc1
commit 0246d68
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/manga_translator/translators/mbart50.py b/manga_translator/translators/mbart50.py
@@ -69,6 +69,7 @@ async def _load(self, from_lang: str, to_lang: str, device: str):
         self.model = MBartForConditionalGeneration.from_pretrained(self._TRANSLATOR_MODEL)
         if self.device != 'cpu':
             self.model.to(self.device)
+        self.model.eval()
         self.tokenizer = AutoTokenizer.from_pretrained(self._TRANSLATOR_MODEL)
 
     async def _unload(self):