"""Diba-Vision processor: prepares images, video and text for the Diba-Vision model (Dibachain)."""
from transformers.models.auto.processing_auto import PROCESSOR_MAPPING

from .configuration_diba_vision import _Base

_BaseProcessor = PROCESSOR_MAPPING[_Base]


class DibaVisionProcessor(_BaseProcessor):
    @classmethod
    def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
        kwargs.setdefault("trust_remote_code", True)
        return super().from_pretrained(pretrained_model_name_or_path, **kwargs)