diff options
Diffstat (limited to 'candle-examples/examples/musicgen/encodec_model.rs')
| -rw-r--r-- | candle-examples/examples/musicgen/encodec_model.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/musicgen/encodec_model.rs b/candle-examples/examples/musicgen/encodec_model.rs index 2ef6f20f..eaf4ca05 100644 --- a/candle-examples/examples/musicgen/encodec_model.rs +++ b/candle-examples/examples/musicgen/encodec_model.rs @@ -142,7 +142,7 @@ impl EncodecEuclideanCodebook { } fn decode(&self, embed_ind: &Tensor) -> Result<Tensor> { - let quantize = Tensor::embedding(embed_ind, &self.embed)?; + let quantize = self.embed.embedding(embed_ind)?; Ok(quantize) } } |