SubstanceSHIFT/SeedVR2-3B

1# // Copyright (c) 2025 Bytedance Ltd. and/or its affiliates2# //3# // Licensed under the Apache License, Version 2.0 (the "License");4# // you may not use this file except in compliance with the License.5# // You may obtain a copy of the License at6# //7# //     http://www.apache.org/licenses/LICENSE-2.08# //9# // Unless required by applicable law or agreed to in writing, software10# // distributed under the License is distributed on an "AS IS" BASIS,11# // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.12# // See the License for the specific language governing permissions and13# // limitations under the License.14 15"""16Distributed package.17"""18 19from .basic import (20    barrier_if_distributed,21    convert_to_ddp,22    get_device,23    get_global_rank,24    get_local_rank,25    get_world_size,26    init_torch,27)28 29__all__ = [30    "barrier_if_distributed",31    "convert_to_ddp",32    "get_device",33    "get_global_rank",34    "get_local_rank",35    "get_world_size",36    "init_torch",37]38