Torch decoder concatenates feature projections as [c4,c3,c2,c1] but MLX was using [c1,c2,c3,c4]. The linear_fuse conv was trained with the Torch order, producing scrambled features. Verified fix gives correlation=1.0 against Torch on identical inputs. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> |
||
|---|---|---|
| .. | ||
| corridorkey_mlx | ||