transformers/utils/test_module/custom_processing.py

7 lines
172 B
Python

from transformers import ProcessorMixin
class CustomProcessor(ProcessorMixin):
feature_extractor_class = "AutoFeatureExtractor"
tokenizer_class = "AutoTokenizer"