Change urls of the models
Browse files
app.py
CHANGED
|
@@ -41,10 +41,10 @@ token_ = os.getenv('HF_TOKEN')
|
|
| 41 |
models_configs = {
|
| 42 |
'base': Config(),
|
| 43 |
'female': Config(
|
| 44 |
-
model_name='nineninesix/
|
| 45 |
),
|
| 46 |
'male': Config(
|
| 47 |
-
model_name='nineninesix/
|
| 48 |
)
|
| 49 |
}
|
| 50 |
|
|
|
|
| 41 |
models_configs = {
|
| 42 |
'base': Config(),
|
| 43 |
'female': Config(
|
| 44 |
+
model_name='nineninesix/kani-tts-450m-0.2-ft',
|
| 45 |
),
|
| 46 |
'male': Config(
|
| 47 |
+
model_name='nineninesix/kani-tts-450m-0.1-ft',
|
| 48 |
)
|
| 49 |
}
|
| 50 |
|
util.py
CHANGED
|
@@ -10,7 +10,7 @@ import os
|
|
| 10 |
|
| 11 |
@dataclass
|
| 12 |
class Config:
|
| 13 |
-
model_name: str = "nineninesix/
|
| 14 |
audiocodec_name: str = "nvidia/nemo-nano-codec-22khz-0.6kbps-12.5fps"
|
| 15 |
device_map: str = "auto"
|
| 16 |
tokeniser_length: int = 64400
|
|
|
|
| 10 |
|
| 11 |
@dataclass
|
| 12 |
class Config:
|
| 13 |
+
model_name: str = "nineninesix/kani-tts-450m-0.1-pt"
|
| 14 |
audiocodec_name: str = "nvidia/nemo-nano-codec-22khz-0.6kbps-12.5fps"
|
| 15 |
device_map: str = "auto"
|
| 16 |
tokeniser_length: int = 64400
|