Input tensor shape error

Keep getting error on three different comfy installs.
```
Error while processing rearrange-reduction pattern "h w c -> c h w".
 Input tensor shape: torch.Size([2, 512, 512, 3]). Additional info: {}.
 Wrong shape: expected 3 dims. Received 4-dim tensor.
```