summaryrefslogtreecommitdiff
path: root/candle-examples/examples/parler-tts/decode.py
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2024-08-19 07:59:51 +0100
committerGitHub <noreply@github.com>2024-08-19 08:59:51 +0200
commit236b29ff1555db82fdb78c1be8741c0ac37859d1 (patch)
tree89c30dcf340e01fce46ed6194877878637cbbcea /candle-examples/examples/parler-tts/decode.py
parent58197e189657b6587a254882abdb232e83e86848 (diff)
downloadcandle-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.py1
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