(self)
| 18 | return used_memory < self.vram_limit |
| 19 | |
| 20 | def offload(self): |
| 21 | if self.state != 0: |
| 22 | self.to(dtype=self.offload_dtype, device=self.offload_device) |
| 23 | self.state = 0 |
| 24 | |
| 25 | def onload(self): |
| 26 | if self.state != 1: |
no test coverage detected