llava/train/sequence_parallel/__init__.py

# Copyright (c) 2025 NVIDIA CORPORATION.
# Licensed under the MIT license.

# Adapted from https://github.com/NVlabs/VILA/tree/main under the Apache 2.0 license.
# LICENSE is in incl_licenses directory.

from .globals import get_pg_manager, set_pg_manager
from .input_utils import extract_local_from_list, extract_local_input_ids, extract_local_position_ids