Update spectrogram and waveform model mapping for TTS/A pipeline (#26114)

update names mapping for spectrogram and waveform models

Update spectrogram and waveform model mapping for TTS/A pipeline (#26114)
update names mapping for spectrogram and waveform models
32ec7345 · Vaibhav Srivastav · GitHub · a9b63ca9 · 32ec7345
Unverified Commit 32ec7345 authored Sep 13, 2023 by Vaibhav Srivastav Committed by GitHub Sep 13, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

utils/update_metadata.py utils/update_metadata.py +2 -2

No files found.
--- 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"),
 ]