Please double check my inference code

[clane9]

Hi @Junebeomstics and SwiFT co-authors. 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.

@ckadirt and I implemented 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!