Simonlob commited on
Commit
e57a7bf
·
verified ·
1 Parent(s): 261d757

Update kanitts/config.py

Browse files
Files changed (1) hide show
  1. kanitts/config.py +1 -1
kanitts/config.py CHANGED
@@ -16,7 +16,7 @@ class AudioConfig:
16
  @dataclass
17
  class ModelConfig:
18
  """Configuration for language model."""
19
- model_name: str = 'nineninesix/kani-tts-450m-0.1-pt'
20
  device_map: str = "auto"
21
  torch_dtype: str = "bfloat16"
22
  max_new_tokens: int = 1200
 
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