(self, vae)
| 19 | TITLE = "LTXV VAE Patcher" |
| 20 | |
| 21 | def patch(self, vae): |
| 22 | from q8_kernels.integration.patch_vae import patch_vae |
| 23 | |
| 24 | vae_model = vae.first_stage_model |
| 25 | vae.memory_used_decode = ( |
| 26 | lambda shape, dtype: shape[1] |
| 27 | * shape[2] |
| 28 | * shape[3] |
| 29 | * shape[4] |
| 30 | * 8 |
| 31 | * 8 |
| 32 | * 8 |
| 33 | * 3 |
| 34 | * dtype_size(dtype) |
| 35 | ) |
| 36 | |
| 37 | patch_vae(vae_model, patch_block=4) |
| 38 | return (vae,) |
nothing calls this directly
no outgoing calls
no test coverage detected