Code bug description

[metezhang]

Hi, thanks a lot for your wonderful work. I found that inside koopmanlab/models/koopmanViT.py there is a module call parameter error in line 215, 221, embed_dim should not be used as a parameter for the call.