def from_native_module(
        module: nn.GroupNorm, process_group: Union[ProcessGroup, List[ProcessGroup]], **kwargs
    ) -> ParallelModule:
        r"""
        Convert a native PyTorch nn.GroupNorm layer to a tensor parallelized layer.
        """

        group_norm_tp = GroupNormTP(
            num_groups=module.num_groups,
            num_channels=module.num_channels,
            eps=module.eps,
            affine=module.affine,
            device=module.weight.device,
            dtype=module.weight.dtype,
            tp_group=process_group,
            weight=module.weight,
            bias=module.bias,
            **kwargs,
        )
        return group_norm_tp

        Convert a native PyTorch nn.GroupNorm layer to a tensor parallelized layer.
        
opensora/models/vae/tensor_parallel.py