Difference between paper and code

Hi,

In https://github.com/yxuansu/MAGIC/blob/main/story_generation/language_model/utlis.py#L266, you say you only use the generated text to calculate similarity. But in the paper, the MAGIC scores are calculated based on the prefix and the generated tokens. Could you explain this difference?

Thx~