Unverified Commit c32b432a authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Fixed a bug where the `pipeline.framework` would actually contain (#10970)

a fully qualified model.

We simply forgot to change the call for this one when this landed:
https://github.com/huggingface/transformers/pull/10888

It's odd that tests didn't catch that. Should we add some ?
(It's a pretty edgy test case, but it does run within the API).
parent e3c8443f
......@@ -528,7 +528,7 @@ class Pipeline(_ScikitCompat):
):
if framework is None:
framework = infer_framework_from_model(model)
framework, model = infer_framework_from_model(model)
self.task = task
self.model = model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment