"examples/run_classifier.py" did not exist on "dee09a40b2c848b9067bf864fe2804fcdd953202"
Unverified Commit 7822cd38 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[tests] make pipelines tests faster with smaller models (#4238)

covers torch and tf. Also fixes a failing @slow test
parent 448c4672
...@@ -1513,7 +1513,7 @@ class TranslationPipeline(Pipeline): ...@@ -1513,7 +1513,7 @@ class TranslationPipeline(Pipeline):
return results return results
# Register all the supported task here # Register all the supported tasks here
SUPPORTED_TASKS = { SUPPORTED_TASKS = {
"feature-extraction": { "feature-extraction": {
"impl": FeatureExtractionPipeline, "impl": FeatureExtractionPipeline,
...@@ -1576,9 +1576,9 @@ SUPPORTED_TASKS = { ...@@ -1576,9 +1576,9 @@ SUPPORTED_TASKS = {
"tf": TFAutoModelWithLMHead if is_tf_available() else None, "tf": TFAutoModelWithLMHead if is_tf_available() else None,
"pt": AutoModelWithLMHead if is_torch_available() else None, "pt": AutoModelWithLMHead if is_torch_available() else None,
"default": { "default": {
"model": {"pt": "bart-large-cnn", "tf": None}, "model": {"pt": "bart-large-cnn", "tf": "t5-small"},
"config": None, "config": None,
"tokenizer": ("bart-large-cnn", {"use_fast": False}), "tokenizer": {"pt": ("bart-large-cnn", {"use_fast": False}), "tf": "t5-small"},
}, },
}, },
"translation_en_to_fr": { "translation_en_to_fr": {
......
This diff is collapsed.
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