Bug: Missing comma in 'reshape'

[squanchess]

The code on line 445 seems to be incorrect.The code on line 445 seems to be incorrect.
Current code:
intermediates = [y.reshape(B, H, W, D -1).permute(0, 4, 1, 2, 3).contiguous() for y in intermediates]

[cclaess]

Hi @squanchess,

Thanks for raising this issue. I have resolved the issues and pushed the fix into the main branch.
Note that the behavior of forward_intermediates has slightly changed.

Best,
Cris