Example in README doesn't work

[Akababa]

I'm not sure, but I think these are the right replacement kwargs:

model = Conformer(dim=dim,
                  dim_head=32,
                  depth=3).to(device)