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