[bug]: kontext-dev pack error #8172
Description
Is there an existing issue for this problem?
- I have searched the existing issues
Operating system
Linux
GPU vendor
Nvidia (CUDA)
GPU model
RTX 3060
GPU VRAM
No response
Version number
Browser
chrome 132
Python dependencies
No response
What happened
given a reference image of size 798 x 818
File "/home/ubuntu/src/InvokeAI/invokeai/app/invocations/flux_denoise.py", line 393, in _run_diffusion
kontext_extension = KontextExtension(
^^^^^^^^^^^^^^^^^
File "/home/ubuntu/src/InvokeAI/invokeai/backend/flux/extensions/kontext_extension.py", line 91, in __init__
self.kontext_latents, self.kontext_ids = self._prepare_kontext()
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ubuntu/src/InvokeAI/invokeai/backend/flux/extensions/kontext_extension.py", line 112, in _prepare_kontext
kontext_latents_packed = pack(kontext_latents_unpacked).to(self._device, self._dtype)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ubuntu/src/InvokeAI/invokeai/backend/flux/sampling_utils.py", line 155, in pack
return rearrange(x, "b c (h ph) (w pw) -> b (h w) (c ph pw)", ph=2, pw=2)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ubuntu/venv/lib/python3.12/site-packages/einops/einops.py", line 600, in rearrange
return reduce(tensor, pattern, reduction="rearrange", **axes_lengths)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ubuntu/venv/lib/python3.12/site-packages/einops/einops.py", line 542, in reduce
raise EinopsError(message + "\n {}".format(e))
einops.EinopsError: Error while processing rearrange-reduction pattern "b c (h ph) (w pw) -> b (h w) (c ph pw)".
Input tensor shape: torch.Size([1, 16, 102, 99]). Additional info: {'ph': 2, 'pw': 2}.
Shape mismatch, can't divide axis of length 99 in chunks of 2
What you expected to happen
unsure if it should crop or pad or rescale
How to reproduce the problem
No response
Additional context
No response
Discord username
No response