From 49c06132df92e51be251469be8ea8137168340f3 Mon Sep 17 00:00:00 2001 From: Rens Date: Wed, 20 May 2020 22:23:21 +0200 Subject: [PATCH] pass on tokenizer to pipeline (#4489) --- src/transformers/convert_graph_to_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/convert_graph_to_onnx.py b/src/transformers/convert_graph_to_onnx.py index 7afe974e08..567e03d721 100644 --- a/src/transformers/convert_graph_to_onnx.py +++ b/src/transformers/convert_graph_to_onnx.py @@ -103,7 +103,7 @@ def load_graph_from_args(framework: str, model: str, tokenizer: Optional[str] = print("Loading pipeline (model: {}, tokenizer: {})".format(model, tokenizer)) # Allocate tokenizer and model - return pipeline("feature-extraction", model=model, framework=framework) + return pipeline("feature-extraction", model=model, tokenizer=tokenizer, framework=framework) def convert_pytorch(nlp: Pipeline, opset: int, output: str, use_external_format: bool):