Spaces:

hugging-apps
/

flexpart

Running on Zero

flexpart / src /models /transformers /modeling_outputs.py

Upload folder using huggingface_hub

820f6f8 verified 3 days ago

175 Bytes

from dataclasses import dataclass
from typing import *
import torch


@dataclass
class Transformer1DModelOutput:
    sample: torch.FloatTensor
    aux_loss: torch.FloatTensor

	from dataclasses import dataclass
	from typing import *
	import torch


	@dataclass
	class Transformer1DModelOutput:
	sample: torch.FloatTensor
	aux_loss: torch.FloatTensor