Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
custom_processing.py 172 Bytes
Newer Older
Sylvain Gugger's avatar
Allow custom code for Processors (#15649)  
Sylvain Gugger committed
1
2
3
4
5
6 
from transformers import ProcessorMixin


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