change input text

huggingface · May 28, 2024 · b458e28 · b458e28
1 parent 62c1f47
commit b458e28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -2279,7 +2279,7 @@ def test_inference_old_onnx_model(self, use_cache):
         tokenizer = get_preprocessor(model_id)
         model = AutoModelForCausalLM.from_pretrained("gpt2")
         tokenizer = get_preprocessor(model_id)
-        text = "This is a sample output"
+        text = "The capital of France is"
         tokens = tokenizer(text, return_tensors="pt")
         onnx_model = ORTModelForCausalLM.from_pretrained(model_id, use_cache=use_cache, use_io_binding=use_cache)