Spaces:
Running
on
Zero
Running
on
Zero
File size: 380 Bytes
d5d0921 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from pydantic import BaseModel
class ChatTTSConfig(BaseModel):
style: str = ""
temperature: float = 0.3
top_p: float = 0.7
top_k: int = 20
prompt1: str = ""
prompt2: str = ""
prefix: str = ""
class InferConfig(BaseModel):
batch_size: int = 4
spliter_threshold: int = 100
# end_of_sentence
eos: str = "[uv_break]"
seed: int = 42
|