Please double check our inference code

[clane9]

Hi @josueortc, @ahof1704, @dvdijk. Thanks for releasing this nice work!

My collaborators and I at MedARC are putting together an evaluation suite to compare current state-of-the-art fMRI foundation models, and we'd like to include your model.

@mihirneal helped us implement an initial inference wrapper for your pretrained model here. Could you please check our implementation to see if we made any mistakes? If you notice any, please feel free to raise an issue or open a PR to fix. Thank you!