How to deal with multi-token target?

[tuanvu92]

Hi,

When using causal tracing to locate the critical layesr, it seems that the target have to be a single token. Do you have any ideas how to apply causal tracing for target that spans several tokens? For example: "Seoul is the capital of South Korea"

[MstarLioning]

Hello, have you set up the environment for the MEMIT paper? I would like to ask where the downloaded covariance weights should be placed. Thank you!