[fix] Seq2Seq model fix for lmi-dist

[maaquib]

Description

Seq2Seq model fix for lmi-dist Pending build of RE with the corresponding backend fix

• If this change is a backward incompatible change, why must this change be made?
• Interesting edge cases to note here