Merge pull request #93 from jason19970210/patch-1

Update README.md for Mamba Block example

Author: tridao

README.md

@@ -44,6 +44,7 @@ Source: [modules/mamba_simple.py](mamba_ssm/modules/mamba_simple.py).
 
 Usage:
 ```
+import torch
 from mamba_ssm import Mamba
 
 batch, length, dim = 2, 64, 16