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> |
||
|---|---|---|
| .. | ||
| convert | ||
| inference | ||
| io | ||
| model | ||
| utils | ||
| __init__.py | ||
| engine.py | ||