mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 01:22:22 +00:00
Update export-onnx.py for vits to support sherpa-onnx. (#1524)
This commit is contained in:
Fangjun Kuang
GitHub
parent
58610b1bf6
commit
29b195a42e
@ -1,3 +1,4 @@
|
|||||||
See https://k2-fsa.github.io/icefall/recipes/TTS/ljspeech/vits.html for detailed tutorials.
|
See https://k2-fsa.github.io/icefall/recipes/TTS/ljspeech/vits.html for detailed tutorials.
|
||||||
|
|
||||||
Training logs, Tensorboard logs, and checkpoints are uploaded to https://huggingface.co/Zengwei/icefall-tts-ljspeech-vits-2023-11-29.
|
Training logs, Tensorboard logs, and checkpoints are uploaded to
|
||||||
|
https://huggingface.co/Zengwei/icefall-tts-ljspeech-vits-2024-02-28
|
||||||
|
|||||||
@ -91,7 +91,7 @@ def add_meta_data(filename: str, meta_data: Dict[str, str]):
|
|||||||
for key, value in meta_data.items():
|
for key, value in meta_data.items():
|
||||||
meta = model.metadata_props.add()
|
meta = model.metadata_props.add()
|
||||||
meta.key = key
|
meta.key = key
|
||||||
meta.value = value
|
meta.value = str(value)
|
||||||
|
|
||||||
onnx.save(model, filename)
|
onnx.save(model, filename)
|
||||||
|
|
||||||
@ -199,10 +199,15 @@ def export_model_onnx(
|
|||||||
)
|
)
|
||||||
|
|
||||||
meta_data = {
|
meta_data = {
|
||||||
"model_type": "VITS",
|
"model_type": "vits",
|
||||||
"version": "1",
|
"version": "1",
|
||||||
"model_author": "k2-fsa",
|
"model_author": "k2-fsa",
|
||||||
"comment": "VITS generator",
|
"comment": "icefall", # must be icefall for models from icefall
|
||||||
|
"language": "English",
|
||||||
|
"voice": "en-us", # Choose your language appropriately
|
||||||
|
"has_espeak": 1,
|
||||||
|
"n_speakers": 1,
|
||||||
|
"sample_rate": 22050, # Must match the real sample rate
|
||||||
}
|
}
|
||||||
logging.info(f"meta_data: {meta_data}")
|
logging.info(f"meta_data: {meta_data}")
|
||||||
|
|
||||||
@ -268,3 +273,144 @@ if __name__ == "__main__":
|
|||||||
formatter = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
|
formatter = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s"
|
||||||
logging.basicConfig(format=formatter, level=logging.INFO)
|
logging.basicConfig(format=formatter, level=logging.INFO)
|
||||||
main()
|
main()
|
||||||
|
|
||||||
|
"""
|
||||||
|
Supported languages.
|
||||||
|
|
||||||
|
LJSpeech is using "en-us" from the second column.
|
||||||
|
|
||||||
|
Pty Language Age/Gender VoiceName File Other Languages
|
||||||
|
5 af --/M Afrikaans gmw/af
|
||||||
|
5 am --/M Amharic sem/am
|
||||||
|
5 an --/M Aragonese roa/an
|
||||||
|
5 ar --/M Arabic sem/ar
|
||||||
|
5 as --/M Assamese inc/as
|
||||||
|
5 az --/M Azerbaijani trk/az
|
||||||
|
5 ba --/M Bashkir trk/ba
|
||||||
|
5 be --/M Belarusian zle/be
|
||||||
|
5 bg --/M Bulgarian zls/bg
|
||||||
|
5 bn --/M Bengali inc/bn
|
||||||
|
5 bpy --/M Bishnupriya_Manipuri inc/bpy
|
||||||
|
5 bs --/M Bosnian zls/bs
|
||||||
|
5 ca --/M Catalan roa/ca
|
||||||
|
5 chr-US-Qaaa-x-west --/M Cherokee_ iro/chr
|
||||||
|
5 cmn --/M Chinese_(Mandarin,_latin_as_English) sit/cmn (zh-cmn 5)(zh 5)
|
||||||
|
5 cmn-latn-pinyin --/M Chinese_(Mandarin,_latin_as_Pinyin) sit/cmn-Latn-pinyin (zh-cmn 5)(zh 5)
|
||||||
|
5 cs --/M Czech zlw/cs
|
||||||
|
5 cv --/M Chuvash trk/cv
|
||||||
|
5 cy --/M Welsh cel/cy
|
||||||
|
5 da --/M Danish gmq/da
|
||||||
|
5 de --/M German gmw/de
|
||||||
|
5 el --/M Greek grk/el
|
||||||
|
5 en-029 --/M English_(Caribbean) gmw/en-029 (en 10)
|
||||||
|
2 en-gb --/M English_(Great_Britain) gmw/en (en 2)
|
||||||
|
5 en-gb-scotland --/M English_(Scotland) gmw/en-GB-scotland (en 4)
|
||||||
|
5 en-gb-x-gbclan --/M English_(Lancaster) gmw/en-GB-x-gbclan (en-gb 3)(en 5)
|
||||||
|
5 en-gb-x-gbcwmd --/M English_(West_Midlands) gmw/en-GB-x-gbcwmd (en-gb 9)(en 9)
|
||||||
|
5 en-gb-x-rp --/M English_(Received_Pronunciation) gmw/en-GB-x-rp (en-gb 4)(en 5)
|
||||||
|
2 en-us --/M English_(America) gmw/en-US (en 3)
|
||||||
|
5 en-us-nyc --/M English_(America,_New_York_City) gmw/en-US-nyc
|
||||||
|
5 eo --/M Esperanto art/eo
|
||||||
|
5 es --/M Spanish_(Spain) roa/es
|
||||||
|
5 es-419 --/M Spanish_(Latin_America) roa/es-419 (es-mx 6)
|
||||||
|
5 et --/M Estonian urj/et
|
||||||
|
5 eu --/M Basque eu
|
||||||
|
5 fa --/M Persian ira/fa
|
||||||
|
5 fa-latn --/M Persian_(Pinglish) ira/fa-Latn
|
||||||
|
5 fi --/M Finnish urj/fi
|
||||||
|
5 fr-be --/M French_(Belgium) roa/fr-BE (fr 8)
|
||||||
|
5 fr-ch --/M French_(Switzerland) roa/fr-CH (fr 8)
|
||||||
|
5 fr-fr --/M French_(France) roa/fr (fr 5)
|
||||||
|
5 ga --/M Gaelic_(Irish) cel/ga
|
||||||
|
5 gd --/M Gaelic_(Scottish) cel/gd
|
||||||
|
5 gn --/M Guarani sai/gn
|
||||||
|
5 grc --/M Greek_(Ancient) grk/grc
|
||||||
|
5 gu --/M Gujarati inc/gu
|
||||||
|
5 hak --/M Hakka_Chinese sit/hak
|
||||||
|
5 haw --/M Hawaiian map/haw
|
||||||
|
5 he --/M Hebrew sem/he
|
||||||
|
5 hi --/M Hindi inc/hi
|
||||||
|
5 hr --/M Croatian zls/hr (hbs 5)
|
||||||
|
5 ht --/M Haitian_Creole roa/ht
|
||||||
|
5 hu --/M Hungarian urj/hu
|
||||||
|
5 hy --/M Armenian_(East_Armenia) ine/hy (hy-arevela 5)
|
||||||
|
5 hyw --/M Armenian_(West_Armenia) ine/hyw (hy-arevmda 5)(hy 8)
|
||||||
|
5 ia --/M Interlingua art/ia
|
||||||
|
5 id --/M Indonesian poz/id
|
||||||
|
5 io --/M Ido art/io
|
||||||
|
5 is --/M Icelandic gmq/is
|
||||||
|
5 it --/M Italian roa/it
|
||||||
|
5 ja --/M Japanese jpx/ja
|
||||||
|
5 jbo --/M Lojban art/jbo
|
||||||
|
5 ka --/M Georgian ccs/ka
|
||||||
|
5 kk --/M Kazakh trk/kk
|
||||||
|
5 kl --/M Greenlandic esx/kl
|
||||||
|
5 kn --/M Kannada dra/kn
|
||||||
|
5 ko --/M Korean ko
|
||||||
|
5 kok --/M Konkani inc/kok
|
||||||
|
5 ku --/M Kurdish ira/ku
|
||||||
|
5 ky --/M Kyrgyz trk/ky
|
||||||
|
5 la --/M Latin itc/la
|
||||||
|
5 lb --/M Luxembourgish gmw/lb
|
||||||
|
5 lfn --/M Lingua_Franca_Nova art/lfn
|
||||||
|
5 lt --/M Lithuanian bat/lt
|
||||||
|
5 ltg --/M Latgalian bat/ltg
|
||||||
|
5 lv --/M Latvian bat/lv
|
||||||
|
5 mi --/M Māori poz/mi
|
||||||
|
5 mk --/M Macedonian zls/mk
|
||||||
|
5 ml --/M Malayalam dra/ml
|
||||||
|
5 mr --/M Marathi inc/mr
|
||||||
|
5 ms --/M Malay poz/ms
|
||||||
|
5 mt --/M Maltese sem/mt
|
||||||
|
5 mto --/M Totontepec_Mixe miz/mto
|
||||||
|
5 my --/M Myanmar_(Burmese) sit/my
|
||||||
|
5 nb --/M Norwegian_Bokmål gmq/nb (no 5)
|
||||||
|
5 nci --/M Nahuatl_(Classical) azc/nci
|
||||||
|
5 ne --/M Nepali inc/ne
|
||||||
|
5 nl --/M Dutch gmw/nl
|
||||||
|
5 nog --/M Nogai trk/nog
|
||||||
|
5 om --/M Oromo cus/om
|
||||||
|
5 or --/M Oriya inc/or
|
||||||
|
5 pa --/M Punjabi inc/pa
|
||||||
|
5 pap --/M Papiamento roa/pap
|
||||||
|
5 piqd --/M Klingon art/piqd
|
||||||
|
5 pl --/M Polish zlw/pl
|
||||||
|
5 pt --/M Portuguese_(Portugal) roa/pt (pt-pt 5)
|
||||||
|
5 pt-br --/M Portuguese_(Brazil) roa/pt-BR (pt 6)
|
||||||
|
5 py --/M Pyash art/py
|
||||||
|
5 qdb --/M Lang_Belta art/qdb
|
||||||
|
5 qu --/M Quechua qu
|
||||||
|
5 quc --/M K'iche' myn/quc
|
||||||
|
5 qya --/M Quenya art/qya
|
||||||
|
5 ro --/M Romanian roa/ro
|
||||||
|
5 ru --/M Russian zle/ru
|
||||||
|
5 ru-cl --/M Russian_(Classic) zle/ru-cl
|
||||||
|
2 ru-lv --/M Russian_(Latvia) zle/ru-LV
|
||||||
|
5 sd --/M Sindhi inc/sd
|
||||||
|
5 shn --/M Shan_(Tai_Yai) tai/shn
|
||||||
|
5 si --/M Sinhala inc/si
|
||||||
|
5 sjn --/M Sindarin art/sjn
|
||||||
|
5 sk --/M Slovak zlw/sk
|
||||||
|
5 sl --/M Slovenian zls/sl
|
||||||
|
5 smj --/M Lule_Saami urj/smj
|
||||||
|
5 sq --/M Albanian ine/sq
|
||||||
|
5 sr --/M Serbian zls/sr
|
||||||
|
5 sv --/M Swedish gmq/sv
|
||||||
|
5 sw --/M Swahili bnt/sw
|
||||||
|
5 ta --/M Tamil dra/ta
|
||||||
|
5 te --/M Telugu dra/te
|
||||||
|
5 th --/M Thai tai/th
|
||||||
|
5 tk --/M Turkmen trk/tk
|
||||||
|
5 tn --/M Setswana bnt/tn
|
||||||
|
5 tr --/M Turkish trk/tr
|
||||||
|
5 tt --/M Tatar trk/tt
|
||||||
|
5 ug --/M Uyghur trk/ug
|
||||||
|
5 uk --/M Ukrainian zle/uk
|
||||||
|
5 ur --/M Urdu inc/ur
|
||||||
|
5 uz --/M Uzbek trk/uz
|
||||||
|
5 vi --/M Vietnamese_(Northern) aav/vi
|
||||||
|
5 vi-vn-x-central --/M Vietnamese_(Central) aav/vi-VN-x-central
|
||||||
|
5 vi-vn-x-south --/M Vietnamese_(Southern) aav/vi-VN-x-south
|
||||||
|
5 yue --/M Chinese_(Cantonese) sit/yue (zh-yue 5)(zh 8)
|
||||||
|
5 yue --/M Chinese_(Cantonese,_latin_as_Jyutping) sit/yue-Latn-jyutping (zh-yue 5)(zh 8)
|
||||||
|
"""
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user