AttributeError: module 'torch_optimizer' has no attribute 'Lion'

I'm facing this problem while trying to access Lion optimizer 
I used

```
import torch_optimizer as optim

optimizer = optim.Lion(ensemble_model.parameters(), lr=1e-4)
```

