Contact prediction for multimeric proteins

[AnjaliSetiya]

Hi all
I'm interested in generating contact maps from the ESM2 attention heads as described in the following github notebook:
https://github.com/facebookresearch/esm/blob/main/examples/contact_prediction.ipynb. But it is not clear to me if I want to give hetero dimer or multimeric protein for predicting contact maps how can I do that. How should the input be given for multi chains for contact map prediction