Concerto-Tiny / processing_concerto.py
bigshanedogg's picture
Upload Concerto-Tiny (transformers port, original code)
b25b243 verified
Raw History Blame Contribute Delete
498 Bytes
"""Concerto processor: ``ConcertoFeatureExtractor`` (the package's inference transform) as its only part."""
from transformers.processing_utils import ProcessorMixin
class ConcertoProcessor(ProcessorMixin):
attributes = ["feature_extractor"]
def __init__(self, feature_extractor=None, **kwargs):
super().__init__(feature_extractor, **kwargs)
def __call__(self, point=None, **kwargs):
return self.feature_extractor(point, **kwargs)
__all__ = ["ConcertoProcessor"]