diff options
Diffstat (limited to 'candle-transformers/src/models/encodec.rs')
-rw-r--r-- | candle-transformers/src/models/encodec.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-transformers/src/models/encodec.rs b/candle-transformers/src/models/encodec.rs index a8d509ce..517b9b1d 100644 --- a/candle-transformers/src/models/encodec.rs +++ b/candle-transformers/src/models/encodec.rs @@ -4,9 +4,8 @@ //! //! Based on implementation from [huggingface/transformers](https://github.com/huggingface/transformers/blob/main/src/transformers/models/encodec/modeling_encodec.py) -#![allow(unused)] use candle::{DType, IndexOp, Layout, Module, Result, Shape, Tensor, D}; -use candle_nn::{conv1d, Conv1d, Conv1dConfig, ConvTranspose1d, VarBuilder}; +use candle_nn::{conv1d, Conv1d, ConvTranspose1d, VarBuilder}; // Encodec Model // https://github.com/huggingface/transformers/blob/main/src/transformers/models/encodec/modeling_encodec.py @@ -226,6 +225,7 @@ impl candle::CustomOp2 for CodebookEncode { } // https://github.com/huggingface/transformers/blob/abaca9f9432a84cfaa95531de4c72334f38a42f2/src/transformers/models/encodec/modeling_encodec.py#L340 +#[allow(unused)] #[derive(Clone, Debug)] pub struct EuclideanCodebook { inited: Tensor, |