Update modeling_wavcoch.py

modeling_wavcoch.py

@@ -11,6 +11,7 @@ import tqdm
 from vector_quantize_pytorch import LFQ
 from huggingface_hub import PyTorchModelHubMixin
 from transformers.tokenization_utils import BatchEncoding
+from transformers import PreTrainedTokenizer
 import chcochleagram
 
 
@@ -135,7 +136,7 @@ class CochleagramTransform:
 ################$$$$$$##################
 
 
-class WavCoch(nn.Module, PyTorchModelHubMixin):
+class WavCoch(nn.Module, PyTorchModelHubMixin, PreTrainedTokenizer):
     def __init__(self, config):
         super().__init__()
         self.N = config.window_size