Spaces:
Sleeping
Sleeping
| import os, json, sys | |
| sys.path.append(".") | |
| from uuid import uuid4 | |
| from typing import List, Dict, Literal, Optional, Any, Union | |
| import urllib.parse | |
| import hashlib | |
| from Synthesizers.base import Base_TTS_Task, ParamItem, init_params_config | |
| def get_params_config(): | |
| try: | |
| with open(os.path.join("Synthesizers/gsv_fast/configs", "params_config.json"), "r", encoding="utf-8") as f: | |
| return init_params_config(json.load(f)) | |
| except: | |
| raise FileNotFoundError("params_config.json not found or invalid.") | |
| params_config = get_params_config() | |
| from pydantic import BaseModel, Field, model_validator | |
| class GSV_TTS_Task(Base_TTS_Task): | |
| # character: Optional[str] = None | |
| # emotion: Optional[str] = None | |
| ref_audio_path: Optional[str] = None | |
| prompt_text: Optional[str] = None | |
| prompt_language: Optional[str] = None | |
| text_language: Optional[str] = None | |
| speaker_id: Optional[int] = None | |
| batch_size: Optional[int] = None | |
| top_k: Optional[int] = None | |
| top_p: Optional[float] = None | |
| temperature: Optional[float] = None | |
| cut_method: Optional[str] = None | |
| max_cut_length: Optional[int] = None | |
| seed: Optional[int] = None | |
| save_temp: Optional[bool] = False | |
| parallel_infer : Optional[bool] = True | |
| repetition_penalty : Optional[float] = 1.35 | |
| # the gsv_fast model only supports 32000 sample rate | |
| sample_rate: int = 32000 | |
| def __init__(self, other_task: Union[BaseModel, dict, None] = None, **data): | |
| data.setdefault('params_config', params_config) | |
| super().__init__(other_task, **data) | |
| def md5(self): | |
| m = hashlib.md5() | |
| if self.task_type == "audio": | |
| m.update(self.src.encode()) | |
| elif self.task_type == "ssml": | |
| m.update(self.ssml.encode()) | |
| elif self.task_type == "text": | |
| m.update(self.text.encode()) | |
| m.update(self.text_language.encode()) | |
| m.update(self.character.encode()) | |
| m.update(str(self.speaker_id).encode()) | |
| m.update(str(self.speed).encode()) | |
| m.update(str(self.top_k).encode()) | |
| m.update(str(self.top_p).encode()) | |
| m.update(str(self.temperature).encode()) | |
| m.update(str(self.cut_method).encode()) | |
| m.update(str(self.emotion).encode()) | |
| return m.hexdigest() | |