def METHOD_NAME(self, config, Q):
    Q = Q[:, :config.query_maxlen]   # NOTE: Candidate generation uses only the query tokens
    embedding_ids, centroid_scores = self.generate_candidates(config, Q)
    return embedding_ids, centroid_scores