get_vocos_model.py 235 B

12345678
  1. from tts_webui.utils.manage_model_state import manage_model_state
  2. @manage_model_state("vocos")
  3. def get_vocos_model(model_name="charactr/vocos-encodec-24khz"):
  4. from vocos import Vocos
  5. return Vocos.from_pretrained(model_name)