Open-Sora/configs/diffusion/inference/plugins/tp.py