From 32ec7345f2d752c294ddf5aff495b657c9cd9d3b Mon Sep 17 00:00:00 2001 From: Vaibhav Srivastav Date: Wed, 13 Sep 2023 15:05:11 +0200 Subject: [PATCH] Update spectrogram and waveform model mapping for TTS/A pipeline (#26114) update names mapping for spectrogram and waveform models --- utils/update_metadata.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/update_metadata.py b/utils/update_metadata.py index 3cda173e5e..f90aae69d5 100644 --- a/utils/update_metadata.py +++ b/utils/update_metadata.py @@ -115,8 +115,8 @@ PIPELINE_TAGS_AND_AUTO_MODELS = [ ("depth-estimation", "MODEL_FOR_DEPTH_ESTIMATION_MAPPING_NAMES", "AutoModelForDepthEstimation"), ("video-classification", "MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING_NAMES", "AutoModelForVideoClassification"), ("mask-generation", "MODEL_FOR_MASK_GENERATION_MAPPING_NAMES", "AutoModelForMaskGeneration"), - ("text-to-audio", "MODEL_FOR_TEXT_TO_SPECTROGRAM_NAMES", "AutoModelForTextToSpectrogram"), - ("text-to-audio", "MODEL_FOR_TEXT_TO_WAVEFORM_NAMES", "AutoModelForTextToWaveform"), + ("text-to-audio", "MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPING_NAMES", "AutoModelForTextToSpectrogram"), + ("text-to-audio", "MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING_NAMES", "AutoModelForTextToWaveform"), ]