Note the CUDA assumption in the Python example

README.md

@@ -62,6 +62,9 @@ pip install mamba_ssm causal_conv1d --no-build-isolation
 
 ## Running the classifier in Python
 
+The example assumes a CUDA GPU (the base checkpoint is stored in bfloat16). On CPU, cast the
+model with `model.float()` and run the forward pass without the autocast block.
+
 ```python
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer