d1-2B / native_lfm3_vl.py
paulpak58's picture mlabonne's picture Aurelien-Lac's picture
d1-2B
97c5704
Raw History Blame Contribute Delete
1.39 kB
"""LFM3-VL's transformers classes (`native_lfm3`): its config, vision tower, picture and video processors."""
from __future__ import annotations
from .native_lfm3 import Lfm3Model # noqa: F401 (registered: the VL builds its text model by config)
try:
from transformers import Lfm3VlConfig, Lfm3VlForConditionalGeneration, Lfm3VlProcessor
except ImportError:
from transformers import AutoConfig, AutoImageProcessor, AutoModel, AutoProcessor, AutoVideoProcessor
from .configuration_lfm3_vl import Lfm3VlConfig, Lfm3VlVisionConfig
from .image_processing_lfm3_vl import Lfm3VlImageProcessor
from .modeling_lfm3_vl import Lfm3VlForConditionalGeneration, Lfm3VlVisionModel
from .processing_lfm3_vl import Lfm3VlProcessor
from .video_processing_lfm3_vl import Lfm3VlVideoProcessor
for config in (Lfm3VlConfig, Lfm3VlVisionConfig):
AutoConfig.register(config.model_type, config, exist_ok=True)
AutoModel.register(Lfm3VlVisionConfig, Lfm3VlVisionModel, exist_ok=True) # the tower, built by config
AutoImageProcessor.register(Lfm3VlConfig, fast_image_processor_class=Lfm3VlImageProcessor, exist_ok=True)
AutoVideoProcessor.register(Lfm3VlConfig, Lfm3VlVideoProcessor, exist_ok=True)
AutoProcessor.register(Lfm3VlConfig, Lfm3VlProcessor, exist_ok=True)
__all__ = ["Lfm3VlConfig", "Lfm3VlForConditionalGeneration", "Lfm3VlProcessor"]