Basic changes to allow loading of local model files (#59)

-modify main,py to not hardcode using DEFAULT_VARIANT

-allow submission of path string to path of variant.yaml file saved in
custom directory in lieu of model variant identifier in CLI to use when
loading model

-addresses #57
This commit is contained in:
teddybear082
2026-01-27 09:42:35 +01:00
committed by GitHub
parent 77cc74a443
commit 2b51649a65
5 changed files with 46 additions and 7 deletions
+14
View File
@@ -19,6 +19,7 @@ This will generate a WAV file `./tts_output.wav` with the default text and voice
- `--text TEXT`: Text to generate (default: "Hello world! I am Kyutai Pocket TTS. I'm fast enough to run on small CPUs. I hope you'll like me.")
- `--voice VOICE`: Path to audio conditioning file (voice to clone) (default: "hf://kyutai/tts-voices/alba-mackenna/casual.wav"). Urls and local paths are supported.
- `--output-path OUTPUT_PATH`: Output path for generated audio (default: "./tts_output.wav")
- `--confg CONFIG_PATH`: Path to custom config.yaml (for loading local model files)
### Generation Parameters
@@ -76,6 +77,19 @@ pocket-tts generate --temperature 1.0
pocket-tts generate --eos-threshold -3.0
```
### Custom Model Config
If you'd like to override the paths from which the models are loaded, you can provide a custom YAML configuration.
Copy pocket_tts/config/b6369a24.yaml and change weights_path:, weights_path_without_voice_cloning: and tokenizer_path: to the paths of the models you want to load.
Then, use the --config option to point to your newly created config.
```bash
# Use a different config
pocket-tts generate --config "C://pocket-tts/my_config.yaml"
```
## Output Format
The generate command always outputs WAV files in the following format:
+12
View File
@@ -18,6 +18,7 @@ This starts a server on `http://localhost:8000` with the default voice model.
- `--host HOST`: Host to bind to (default: "localhost")
- `--port PORT`: Port to bind to (default: 8000)
- `--reload`: Enable auto-reload for development
- `--config`: Path to a custom config .yaml
## Examples
@@ -40,6 +41,17 @@ pocket-tts serve --voice "hf://kyutai/tts-voices/jessica-jian/casual.wav"
# Use local voice file
pocket-tts serve --voice "./my_voice.wav"
```
### Custom Model Config
If you'd like to override the paths from which the models are loaded, you can provide a custom YAML configuration.
Copy pocket_tts/config/b6369a24.yaml and change weights_path:, weights_path_without_voice_cloning: and tokenizer_path: to the paths of the models you want to load.
Then, use the --config option to point to your newly created config.
```bash
# Use a different config
pocket-tts serve --config "C://pocket-tts/my_config.yaml"
```
## Web Interface