Turn word embedding to word id (using T5 decoder)
i can get word embedding byoutput = model(input_ids=input_ids)but how can i turn output back to input_idsor something that input an embedding an output the highest prob of the word idthx