(self, *args, **kwargs)
| 569 | assert 'adapter' in config, 'Block swapping only works when training LoRA' |
| 570 | # Don't automatically move to GPU, we'll do that ourselves. |
| 571 | def to(self, *args, **kwargs): |
| 572 | pass |
| 573 | deepspeed.pipe.PipelineModule.to = to |
| 574 | model.enable_block_swap(blocks_to_swap) |
| 575 |
nothing calls this directly
no outgoing calls
no test coverage detected