Method forward

tests/unit/test_modernbert_embed.py:26–28

(self, input_ids, attention_mask, return_dict=True)

        config = types.SimpleNamespace(hidden_size=8)

        def forward(self, input_ids, attention_mask, return_dict=True):
            batch = input_ids.size(0)
            return types.SimpleNamespace(last_hidden_state=torch.zeros((batch, 1, 8)))

    return DummyModel()