From 0c49e95dfb5f25a17340ff1b690a4b4f6cd0e2d1 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Tue, 27 Feb 2024 22:59:40 +0100
Subject: Encodec model. (#1771)

* 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.
---
 candle-nn/src/rnn.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'candle-nn')

diff --git a/candle-nn/src/rnn.rs b/candle-nn/src/rnn.rs
index 9f144cca..07795eda 100644
--- a/candle-nn/src/rnn.rs
+++ b/candle-nn/src/rnn.rs
@@ -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)
     }
 }
 
-- 
cgit v1.2.3