25 lines
714 B
Python
25 lines
714 B
Python
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||
|
|
"""InternVLA-A1 diffusion model components."""
|
||
|
|
|
||
|
|
from vllm_omni.diffusion.models.internvla_a1.config import (
|
||
|
|
InternVLAA1Config,
|
||
|
|
InternVLAA1TrainMetadata,
|
||
|
|
)
|
||
|
|
from vllm_omni.diffusion.models.internvla_a1.model_internvla_a1 import (
|
||
|
|
InternVLAA1,
|
||
|
|
InternVLAA1Policy,
|
||
|
|
)
|
||
|
|
from vllm_omni.diffusion.models.internvla_a1.pipeline_internvla_a1 import (
|
||
|
|
InternVLAA1Pipeline,
|
||
|
|
get_internvla_a1_post_process_func,
|
||
|
|
)
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"InternVLAA1",
|
||
|
|
"InternVLAA1Config",
|
||
|
|
"InternVLAA1Pipeline",
|
||
|
|
"InternVLAA1TrainMetadata",
|
||
|
|
"InternVLAA1Policy",
|
||
|
|
"get_internvla_a1_post_process_func",
|
||
|
|
]
|