[Pytorch] Generative retrieval model using semantic IDs from "Recommender Systems with Generative Retrieval"