Seperate `class SSM` and `class MambaBlock`.

Add installation instructions to README

Create model.py2

Update model.py

Enable using the model on GPU