src/transformers/models/oneformer/modeling_oneformer.py/OneFormerForUniversalSegmentation/get_loss
class OneFormerForUniversalSegmentation:    def get_loss(self, loss_dict: Dict[str, Tensor]) -> Tensor:
        return sum(loss_dict.values())