diff options
Diffstat (limited to 'candle-examples/examples/musicgen/encodec_model.rs')
| -rw-r--r-- | candle-examples/examples/musicgen/encodec_model.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/musicgen/encodec_model.rs b/candle-examples/examples/musicgen/encodec_model.rs index eaf4ca05..9c966497 100644 --- a/candle-examples/examples/musicgen/encodec_model.rs +++ b/candle-examples/examples/musicgen/encodec_model.rs @@ -1,6 +1,7 @@ use crate::nn::{conv1d, conv1d_weight_norm, Conv1d, Conv1dConfig, VarBuilder}; use anyhow::Result; use candle::{DType, IndexOp, Tensor}; +use candle_nn::Module; // Encodec Model // https://github.com/huggingface/transformers/blob/main/src/transformers/models/encodec/modeling_encodec.py |