src/models/transformers/modeling_outputs.py

from dataclasses import dataclass

import torch


@dataclass
class Transformer1DModelOutput:
    sample: torch.FloatTensor