Commit Graph
8 Commits
Author SHA1 Message Date
facok cc5a8cbd6f Pad odd spatial dims in patchify for LTXV compatibility
LTXV can produce latents with odd H or W (e.g. 11x20) which can't be
divided into 2x2 patches. Patchify now pads odd dims to even using
replicate padding before rearranging, and unpatchify crops back to the
original size. This keeps patch_size=2 consistent between calibration
and inference for all models.
2026-03-21 21:39:46 +08:00
facok 4948395ac4 Skip intervention gracefully for incompatible latent formats (LTXAV)
LTXAV uses a flattened 1D latent layout (e.g. [1, 1, 466048]) where
spatial H/W < 2, making 2x2 patchification impossible. patchify() now
returns None for such formats, and all hooks skip intervention cleanly.
2026-03-21 20:40:53 +08:00
facok 5154bae137 Handle 3D tensors (LTXAV) in patchify/unpatchify 2026-03-21 20:34:40 +08:00
facok 0d82606fb0 Support 5D video VAE latents (Wan) in patchify/unpatchify
Video VAEs output [B, C, T, H, W]. Patchify now merges T into batch,
processes all frames, and unpatchify restores the 5D shape.
Updated all callers to pass extra_shape through.
2026-03-21 16:19:03 +08:00
facok 8c1d987b4f Fix patchify for video VAEs: slice first frame when ndim > 4 2026-03-21 15:33:32 +08:00
facok 6dd89e55e7 Handle 5D video VAE output (Wan) by squeezing temporal dimension 2026-03-21 15:21:51 +08:00
facok b3b08761c3 Auto-detect channel count in unpatchify for FLUX2 (128ch) support 2026-03-21 14:57:52 +08:00
facok 534bc19033 Implement LCS plugin for training-free color control in FLUX
Implements "The Latent Color Subspace" (arXiv:2603.12261v1, ICML 2026)
as a ComfyUI plugin with V3 API + V2 backward compatibility.

6 nodes across 3 categories:
- LCS/calibration: LCSCalibrate (PCA from VAE), LCSLoadData (cache load)
- LCS/intervention: LCSColorIntervene (single color), LCSColorBatch (multi)
- LCS/observe: LCSPreviewColors (latent color map), LCSStepObserver

Core modules: patchify (16ch->64d patches), bicone HSL<->LCS mapping,
alpha_t/beta_t tables (51 entries from Appendix F), PCA calibration,
and timestep normalization with residual preservation.
2026-03-17 01:43:26 +08:00