Commit
·
836b884
1
Parent(s):
84d74c3
Update processing_moss_tts.py: Add audio tokenizer mapping to processing_utils for moss audio tokenizer
Browse files- 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,
|