26 lines
670 B
Python
26 lines
670 B
Python
from typing import Type
|
|
|
|
from transformers import (
|
|
AutoImageProcessor,
|
|
AutoProcessor,
|
|
BaseImageProcessor,
|
|
PretrainedConfig,
|
|
ProcessorMixin,
|
|
)
|
|
|
|
|
|
def register_image_processor(
|
|
config: Type[PretrainedConfig], image_processor: Type[BaseImageProcessor]
|
|
):
|
|
"""
|
|
register customized hf image processor while removing hf impl
|
|
"""
|
|
AutoImageProcessor.register(config, None, image_processor, None, exist_ok=True)
|
|
|
|
|
|
def register_processor(config: Type[PretrainedConfig], processor: Type[ProcessorMixin]):
|
|
"""
|
|
register customized hf processor while removing hf impl
|
|
"""
|
|
AutoProcessor.register(config, processor, exist_ok=True)
|