🚀[FEA]: Enable 3D patching/multi-diffusion for CorrDiff-like diffusion models

[CharlelieLrt]

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Low (would be nice)

Please provide a clear description of problem you would like to solve.

Some users reported a need for 3D multi-diffusion models. This will require a refactoring of CorrDiff patching logic to enable 3D patches.

Describe any alternatives you have considered

No response