MIDI-3D / midi /models /transformers /modeling_outputs.py
huanngzh's picture
init
c4ea5b9
raw
history blame
123 Bytes
from dataclasses import dataclass
import torch
@dataclass
class Transformer1DModelOutput:
sample: torch.FloatTensor