(self)
| 26 | assert all(isinstance(l, nn.Module) for l in layers) |
| 27 | |
| 28 | def test_get_hidden_state(self): |
| 29 | model = TinyModel() |
| 30 | adapter = get_adapter(model) |
| 31 | x = torch.randn(2, 16, 64) |
| 32 | layer = adapter.get_layers(model)[0] |
| 33 | out = layer(x) |
| 34 | hidden = adapter.get_hidden_state(out) |
| 35 | assert hidden.shape == (2, 16, 64) |
| 36 | |
| 37 | def test_get_final_norm(self): |
| 38 | model = TinyModel() |
nothing calls this directly
no test coverage detected