(self)
| 74 | vae_decoder.to_cuda() |
| 75 | |
| 76 | def _restore_vae(self) -> None: |
| 77 | vae_decoder = self._vae_decoder |
| 78 | if vae_decoder is None or not self._saved_vae_cpu_offload: |
| 79 | return |
| 80 | vae_decoder.cpu_offload = True |
| 81 | if hasattr(vae_decoder, "to_cpu"): |
| 82 | vae_decoder.to_cpu() |
| 83 | self._saved_vae_cpu_offload = False |
| 84 | |
| 85 | @property |
| 86 | def is_async(self) -> bool: |