Update evaluate-accuracy.py | Fixes #2008

mlcommons · Jan 1, 2025 · 5d5da3c · 5d5da3c
1 parent 308147a
commit 5d5da3c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/language/mixtral-8x7b/evaluate-accuracy.py b/language/mixtral-8x7b/evaluate-accuracy.py
@@ -121,6 +121,7 @@ def main():
     checkpoint_path = args.checkpoint_path
     metric = evaluate.load("rouge")
     nltk.download("punkt")
+    nltk.download("punkt_tab")
 
     tokenizer = AutoTokenizer.from_pretrained(
         checkpoint_path,