Simonlob commited on
Commit
581f616
·
verified ·
1 Parent(s): 58fd339

Update kanitts/config.py

Browse files
Files changed (1) hide show
  1. kanitts/config.py +2 -2
kanitts/config.py CHANGED
@@ -16,10 +16,10 @@ class AudioConfig:
16
  @dataclass
17
  class ModelConfig:
18
  """Configuration for language model."""
19
- model_name: str = 'nineninesix/kani-tts-450m-0.3-pt'
20
  device_map: str = "auto"
21
  torch_dtype: str = "bfloat16"
22
- max_new_tokens: int = 1200
23
  temperature: float = 0.6
24
  top_p: float = 0.95
25
  repetition_penalty: float = 1.1
 
16
  @dataclass
17
  class ModelConfig:
18
  """Configuration for language model."""
19
+ model_name: str = 'nineninesix/kani-tts-400m-0.3-pt'
20
  device_map: str = "auto"
21
  torch_dtype: str = "bfloat16"
22
+ max_new_tokens: int = 1400
23
  temperature: float = 0.6
24
  top_p: float = 0.95
25
  repetition_penalty: float = 1.1