hub / github.com/kyegomez/BitNet / test_bitnet_transformer_embed

Function test_bitnet_transformer_embed

tests/tests.py:198–201 · view source on GitHub ↗

(bitnet_model)

Source from the content-addressed store, hash-verified

196
197
198	def test_bitnet_transformer_embed(bitnet_model):
199	tokens = torch.randint(0, 20000, (1, 512))
200	embedded = bitnet_model.emb(tokens)
201	assert embedded.shape == (1, 512, 512)
202
203
204	@pytest.mark.parametrize(

nothing calls this directly

no outgoing calls

no test coverage detected