(self)
| 53 | self.state = 0 |
| 54 | |
| 55 | def offload(self): |
| 56 | if self.state == 1 and (self.offload_dtype != self.onload_dtype or self.offload_device != self.onload_device): |
| 57 | self.to(dtype=self.offload_dtype, device=self.offload_device) |
| 58 | self.state = 0 |
| 59 | |
| 60 | def onload(self): |
| 61 | if self.state == 0 and (self.offload_dtype != self.onload_dtype or self.offload_device != self.onload_device): |