(bitnet_model)
| 182 | |
| 183 | |
| 184 | def test_bitnet_transformer_forward(bitnet_model): |
| 185 | tokens = torch.randint(0, 20000, (1, 512)) |
| 186 | logits = bitnet_model(tokens) |
| 187 | assert logits.shape == (1, 20000) |
| 188 | |
| 189 | |
| 190 | def test_parallel_transformer_block_masking(random_tensor): |
nothing calls this directly
no test coverage detected