diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-28 06:49:03 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-28 06:49:03 +0100 |
| commit | d0aca6c3c629114c70b4af1d5c44f15857368636 (patch) | |
| tree | e99951a48f9e4eea994d7c4bf3446319c27882ac /candle-transformers/src/models/encodec.rs | |
| parent | 15e864414982b9f6de273ab8e984f700f58bbefe (diff) | |
| download | candle-d0aca6c3c629114c70b4af1d5c44f15857368636.tar.gz candle-d0aca6c3c629114c70b4af1d5c44f15857368636.tar.bz2 candle-d0aca6c3c629114c70b4af1d5c44f15857368636.zip | |
Encodec encoding demo. (#1775)
Diffstat (limited to 'candle-transformers/src/models/encodec.rs')
| -rw-r--r-- | candle-transformers/src/models/encodec.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/candle-transformers/src/models/encodec.rs b/candle-transformers/src/models/encodec.rs index d3b26e1e..1316098b 100644 --- a/candle-transformers/src/models/encodec.rs +++ b/candle-transformers/src/models/encodec.rs @@ -283,7 +283,8 @@ impl VectorQuantization { } pub fn encode(&self, xs: &Tensor) -> Result<Tensor> { - self.codebook.encode_slow(xs) + let xs = xs.transpose(1, 2)?; + self.codebook.encode_slow(&xs) } pub fn decode(&self, embed_ind: &Tensor) -> Result<Tensor> { |