vllm_omni.diffusion.models.internvla_a1.pipeline_internvla_a1 ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

InternVLA-A1 pipeline wrapper for the policy implementation.

config = _build_config(od_config)

enable_warmup = (
    bool(enable_warmup)
    if isinstance(enable_warmup, bool)
    else False
)

model_dir = model

od_config = od_config

policy = _initialize_policy()

prefix = prefix

processor_model_name = str(
    get("processor_model_name", DEFAULT_QWEN3_VL_MODEL)
)

strict_load = bool(get('strict_load', False))

forward(req: OmniDiffusionRequest) -> DiffusionOutput

has_real_checkpoint() -> bool

runtime_mode() -> str

get_internvla_a1_post_process_func(
    od_config: OmniDiffusionConfig,
)