Skip to content

Commit e1ac289

Browse files
pcuencaVaibhavs10
andauthored
Expose audio-text-to-text task (#1758)
Still [not showing up](https://huggingface.co/tasks/audio-text-to-text) despite the upgrade to the tasks package being deployed yesterday. I'm not sure if we also need a mapping to transformers in `TASKS_MODEL_LIBRARIES`. --------- Co-authored-by: vb <vaibhavs10@gmail.com>
1 parent b805f27 commit e1ac289

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

packages/tasks/src/tasks/index.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ import { PIPELINE_DATA } from "../pipelines.js";
33

44
import anyToAny from "./any-to-any/data.js";
55
import audioClassification from "./audio-classification/data.js";
6+
import audioTextToText from "./audio-text-to-text/data.js";
67
import audioToAudio from "./audio-to-audio/data.js";
78
import automaticSpeechRecognition from "./automatic-speech-recognition/data.js";
89
import documentQuestionAnswering from "./document-question-answering/data.js";
@@ -121,7 +122,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
121122
"audio-classification": ["speechbrain", "transformers", "transformers.js"],
122123
"audio-to-audio": ["asteroid", "fairseq", "speechbrain"],
123124
"automatic-speech-recognition": ["espnet", "nemo", "speechbrain", "transformers", "transformers.js"],
124-
"audio-text-to-text": [],
125+
"audio-text-to-text": ["transformers"],
125126
"depth-estimation": ["transformers", "transformers.js"],
126127
"document-question-answering": ["transformers", "transformers.js"],
127128
"feature-extraction": ["sentence-transformers", "transformers", "transformers.js"],
@@ -205,7 +206,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
205206
"any-to-any": getData("any-to-any", anyToAny),
206207
"audio-classification": getData("audio-classification", audioClassification),
207208
"audio-to-audio": getData("audio-to-audio", audioToAudio),
208-
"audio-text-to-text": getData("audio-text-to-text", placeholder),
209+
"audio-text-to-text": getData("audio-text-to-text", audioTextToText),
209210
"automatic-speech-recognition": getData("automatic-speech-recognition", automaticSpeechRecognition),
210211
"depth-estimation": getData("depth-estimation", depthEstimation),
211212
"document-question-answering": getData("document-question-answering", documentQuestionAnswering),

0 commit comments

Comments
 (0)