fix mc loading

ceae85ad · thomwolf · 71883b6d · ceae85ad
Commit ceae85ad authored Dec 20, 2019 by thomwolf
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

transformers/pipelines.py transformers/pipelines.py +4 -0

No files found.
--- a/transformers/pipelines.py
+++ b/transformers/pipelines.py
@@ -891,6 +891,10 @@ def pipeline(task: str, model: Optional = None,
    if isinstance(config, str):
        config = AutoConfig.from_pretrained(config)

+    # Instantiate modelcard if needed
+    if isinstance(modelcard, str):
+        modelcard = ModelCard.from_pretrained(modelcard)
+
    # Instantiate model if needed
    if isinstance(model, str):
        # Handle transparent TF/PT model conversion