Update tasks.py

huggingface · Nov 26, 2023 · e7b29f1 · e7b29f1
1 parent d68fd15
commit e7b29f1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",
         }