Naive inference and updating now works for rerank model