Support context window for PiD and fix lq_latent rounding

[kijai]

Context windows are a way to handle the aspect ratio limitations of the model, also noticed mismatch with some dimensions due to rounding differences, so two fixes:

• Context window slicing needed to account for the lq_latent scaling

• PiD upscaling failed with a tensor size mismatch when the output width or height wasn't a multiple of 16. The two halves of the model rounded the size differently, so they ended up one row/column apart and couldn't be combined.

Also in this situation almost halves peak VRAM use:

Context window settings (easier to use node specific to this model can be added later):

[coderabbitai]

Caution

Review failed

An error occurred during the review process. Please try again later.

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[makisekurisu-jp]

Context windows are a way to handle the aspect ratio limitations of the model, also noticed mismatch with some dimensions due to rounding differences, so two fixes:

• Context window slicing needed to account for the lq_latent scaling
• PiD upscaling failed with a tensor size mismatch when the output width or height wasn't a multiple of 16. The two halves of the model rounded the size differently, so they ended up one row/column apart and couldn't be combined.

Also in this situation almost halves peak VRAM use:

Context window settings (easier to use node specific to this model can be added later):

ValueError: Input latent has 3 channels, this model variant expects 16. Flux1/SD3 = 16 channels, Flux2 = 128 channels.

[kijai]

Context windows are a way to handle the aspect ratio limitations of the model, also noticed mismatch with some dimensions due to rounding differences, so two fixes:

• Context window slicing needed to account for the lq_latent scaling
• PiD upscaling failed with a tensor size mismatch when the output width or height wasn't a multiple of 16. The two halves of the model rounded the size differently, so they ended up one row/column apart and couldn't be combined.

Also in this situation almost halves peak VRAM use:

Context window settings (easier to use node specific to this model can be added later):

ValueError: Input latent has 3 channels, this model variant expects 16. Flux1/SD3 = 16 channels, Flux2 = 128 channels.

PiD only supports flux or sd3 latents, you can't feed pixel space into it like that.