add pegasus to cli tests

huggingface · Dec 21, 2022 · f38ff12 · f38ff12
1 parent 34d12bf
commit f38ff12
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py
@@ -41,6 +41,7 @@
     "mobilevit": "apple/mobilevit-small",
     "mt5": "lewtun/tiny-random-mt5",
     # "owlvit": "google/owlvit-base-patch32",
+    "pegasus": "hf-internal-testing/tiny-random-PegasusModel",
     "perceiver": {
         "hf-internal-testing/tiny-random-language_perceiver": ["masked-lm", "sequence-classification"],
         "hf-internal-testing/tiny-random-vision_perceiver_conv": ["image-classification"],