Should return a tuple for serialization (#6061)

huggingface · Jul 28, 2020 · 7214954 · 7214954
1 parent 7a68d40
commit 7214954
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/pipelines.py b/src/transformers/pipelines.py
@@ -2122,6 +2122,6 @@ def pipeline(
                 "Model might be a PyTorch model (ending with `.bin`) but PyTorch is not available. "
                 "Trying to load the model with Tensorflow."
             )
-        model = model_class.from_pretrained(model, config=config, **model_kwargs)
+        model = model_class.from_pretrained(model, config=config, return_tuple=True, **model_kwargs)
 
     return task_class(model=model, tokenizer=tokenizer, modelcard=modelcard, framework=framework, task=task, **kwargs)