torch.save(model.state_dict(), "vjepa_model.pth")