()
| 206 | # ModelOffloader Tests |
| 207 | @pytest.fixture |
| 208 | def model_offloader(): |
| 209 | device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') |
| 210 | blocks_to_swap = 2 |
| 211 | blocks = SimpleModel(4).blocks |
| 212 | return ModelOffloader( |
| 213 | blocks=blocks, |
| 214 | blocks_to_swap=blocks_to_swap, |
| 215 | device=device, |
| 216 | debug=False |
| 217 | ) |
| 218 | |
| 219 | |
| 220 | def test_model_offloader_init(model_offloader): |
nothing calls this directly
no test coverage detected