Merge pull request #93 from jason19970210/patch-1

Update README.md for Mamba Block example
state-spaces · Jan 5, 2024 · da2626b · da2626b
2 parents fb7b531 + 25c9b9a
commit da2626b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/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