If the indices are defined by the vocabulary, I don’t think it’s a good idea to cut them somehow as this part of the vocabulary wouldn’t be used during training.
Instead of changing the input I would recommend to adapt the num_embeddings in the emnedding layer to match the number of words (indices).