Spaces:
Running on Zero
Running on Zero
File size: 123 Bytes
9d7cf7f | 1 2 3 4 5 6 7 8 9 | from dataclasses import dataclass
import torch
@dataclass
class Transformer1DModelOutput:
sample: torch.FloatTensor
|