fix test path

huggingface · May 1, 2020 · 2959086 · 2959086
1 parent 9604648
commit 2959086
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_modeling_reformer.py b/tests/test_modeling_reformer.py
@@ -1106,8 +1106,8 @@ def test_lsh_lm_model_grad(self):
 
     @slow
     def test_pretrained_generate_crime_and_punish(self):
-        model = ReformerModelWithLMHead.from_pretrained("google/reformer-crime-and-punish").to(torch_device)
-        tokenizer = ReformerTokenizer.from_pretrained("google/reformer-crime-and-punish")
+        model = ReformerModelWithLMHead.from_pretrained("google/reformer-crime-and-punishment").to(torch_device)
+        tokenizer = ReformerTokenizer.from_pretrained("google/reformer-crime-and-punishment")
         model.eval()
 
         input_ids = tokenizer.encode("A few months later", return_tensors="pt").to(torch_device)