Li-Ruixiao commited on
Commit
836b884
·
1 Parent(s): 84d74c3

Update processing_moss_tts.py: Add audio tokenizer mapping to processing_utils for moss audio tokenizer

Browse files
Files changed (1) hide show
  1. processing_moss_tts.py +4 -0
processing_moss_tts.py CHANGED
@@ -20,6 +20,10 @@ from pathlib import Path
20
  import re
21
  import torchaudio
22
 
 
 
 
 
23
  import torch
24
  from transformers import (
25
  PreTrainedTokenizerBase,
 
20
  import re
21
  import torchaudio
22
 
23
+ from transformers import processing_utils
24
+
25
+ processing_utils.MODALITY_TO_BASE_CLASS_MAPPING["audio_tokenizer"] = "PreTrainedModel"
26
+
27
  import torch
28
  from transformers import (
29
  PreTrainedTokenizerBase,