diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index ec9af6dffa4..181f2bbadb8 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -182,6 +182,7 @@ class TasksManager: "text2text-generation": "AutoModelForSeq2SeqLM", "text-classification": "AutoModelForSequenceClassification", "token-classification": "AutoModelForTokenClassification", + "zero-shot-audio-classification": "AutoModel", "zero-shot-image-classification": "AutoModelForZeroShotImageClassification", "zero-shot-object-detection": "AutoModelForZeroShotObjectDetection", }