MCPcopy Create free account
hub / github.com/OpenSparseLLMs/MoM / MomOutputWithPast

Class MomOutputWithPast

mom/models/mom/modeling_mom.py:239–240  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

237
238@dataclass
239class MomOutputWithPast(BaseModelOutputWithPast):
240 router_logits: Optional[Tuple[torch.FloatTensor, ...]] = None
241
242
243class MomModel(MomPreTrainedModel):

Callers 1

forwardMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected