diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-08-19 07:59:51 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-08-19 08:59:51 +0200 |
commit | 236b29ff1555db82fdb78c1be8741c0ac37859d1 (patch) | |
tree | 89c30dcf340e01fce46ed6194877878637cbbcea /candle-examples/examples/parler-tts/decode.py | |
parent | 58197e189657b6587a254882abdb232e83e86848 (diff) | |
download | candle-236b29ff1555db82fdb78c1be8741c0ac37859d1.tar.gz candle-236b29ff1555db82fdb78c1be8741c0ac37859d1.tar.bz2 candle-236b29ff1555db82fdb78c1be8741c0ac37859d1.zip |
Add the DAC model. (#2433)
* Add the DAC model.
* More quantization support.
* Handle DAC decoding.
* Plug the DAC decoding in parler-tts.
Diffstat (limited to 'candle-examples/examples/parler-tts/decode.py')
-rw-r--r-- | candle-examples/examples/parler-tts/decode.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/parler-tts/decode.py b/candle-examples/examples/parler-tts/decode.py index b79ebda1..8942d32e 100644 --- a/candle-examples/examples/parler-tts/decode.py +++ b/candle-examples/examples/parler-tts/decode.py @@ -5,6 +5,7 @@ from parler_tts import DACModel tensors = load_file("out.safetensors") dac_model = DACModel.from_pretrained("parler-tts/dac_44khZ_8kbps") +print(dac_model.model) output_ids = tensors["codes"][None, None] print(output_ids, "\n", output_ids.shape) batch_size = 1 |