Spaces:
Runtime error
Runtime error
Update kanitts/config.py
Browse files- 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-
|
| 20 |
device_map: str = "auto"
|
| 21 |
torch_dtype: str = "bfloat16"
|
| 22 |
-
max_new_tokens: int =
|
| 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
|