How to use not-lain/swarmformer-base with SwarmFormer:
from swarmformer import SwarmFormerModel model = SwarmFormerModel.from_pretrained("not-lain/swarmformer-base")
