MCPcopy Create free account
hub / github.com/MatrixTeam-AI/RAIN / TemporalTransformer3DModelOutput

Class TemporalTransformer3DModelOutput

src/models/motion_module.py:23–24  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21
22@dataclass
23class TemporalTransformer3DModelOutput(BaseOutput):
24 sample: torch.FloatTensor
25
26
27if is_xformers_available():

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected