Hey, are you able to post a repro of how to produce your traced model? Either a link to a .pt file or (even better) the code you use to trace and save the model.
From what I can tell the input to this module forw_sentence is being used as the indices for the embedding operation (which, as in the error, must be of type Long instead of Float), but it’s hard to tell what the actual error is without the full repro.