yangkun.zhuandwangbolun
0016006974
Signed-off-by: zhuyangkun <zyk_120403228@163.com> Signed-off-by: bl-wang1 <918395762@qq.com> Co-authored-by: wangbolun <918395762@qq.com>
26 lines
714 B
Python
26 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",
|
|
]
|