huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Encodec model. (#1771)

Browse Source 
* Encodec model.

* Fixes.

* Add the padding functions.

* Get the LSTM bit to work.

* Get the encodec model to generate some tokens (decoder only for now).

* Minor tweak.

* Minor tweak.
This commit is contained in:
Laurent Mazare
2024-02-27 22:59:40 +01:00
committed by GitHub
parent 205767f9de
commit 0c49e95dfb
7 changed files with 834 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-nn/src/rnn.rs
View File

@ -197,7 +197,7 @@ impl RNN for LSTM {
fn states_to_tensor(&self, states: &[Self::State]) -> Result<Tensor> {
let states = states.iter().map(|s| s.h.clone()).collect::<Vec<_>>();
Tensor::cat(&states, 1)
Tensor::stack(&states, 1)
}
}