File size: 1,542 Bytes
66e3394
a82e5d8
 
 
 
f5eae9c
 
66e3394
f5eae9c
66e3394
a82e5d8
 
 
ba3bbc7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
66e3394
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from .bridge import CortexVisionBridge, TransformersImageCaptioner, VisionObservation, build_default_bridge
try:
    from .native import NativeFusionResult, NativeVisionFusion, build_native_fusion
except ImportError:
    pass

__all__ = [
    "CortexVisionBridge", "TransformersImageCaptioner", "VisionObservation", "build_default_bridge",
]

if "NativeVisionFusion" in globals():
    __all__ += ["NativeFusionResult", "NativeVisionFusion", "build_native_fusion"]

try:
    from .train import (
        ProjectorTrainingConfig,
        TrainingReport,
        build_fusion_pipeline,
        run_training,
        save_projector,
        train_step,
    )

    __all__ += [
        "ProjectorTrainingConfig", "TrainingReport", "build_fusion_pipeline",
        "run_training", "save_projector", "train_step",
    ]
except ImportError:
    # Training helpers require the optional dataset/transformers stack.
    pass

try:
    from .core import (
        ImageProcessor,
        MultimodalFusion,
        MultimodalInput,
        PrefixFusion,
        VisionAdapter,
        VisionConfig,
        VisionEncoder,
        VisionProjector,
        build_vision_adapter,
    )
    __all__ += [
        "ImageProcessor", "MultimodalFusion", "MultimodalInput", "PrefixFusion",
        "VisionAdapter", "VisionConfig", "VisionEncoder", "VisionProjector",
        "build_vision_adapter",
    ]
except ImportError:
    # The caption-to-context bridge can be inspected and unit-tested without
    # importing the optional PyTorch tensor adapter.
    pass