Update tests/onnxruntime/test_modeling.py

Co-authored-by: Ella Charlaix <[email protected]>
huggingface · May 28, 2024 · 4a6bda2 · 4a6bda2
1 parent 7a41a42
commit 4a6bda2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -2276,7 +2276,7 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
     @parameterized.expand([(False,), (True,)])
     def test_inference_old_onnx_model(self, use_cache):
         model_id = "optimum/gpt2"
-        tokenizer = get_preprocessor("gpt2")
+        tokenizer = AutoTokenizer.from_pretrained("gpt2")
         model = AutoModelForCausalLM.from_pretrained("gpt2")
 
         text = "The capital of France is"