Update README.md

README.md

@@ -35,7 +35,7 @@ Since this is a base model the IKM dataset greatly affects the output. The IKM d
 
 ## Inference
 
-```
+```py
 !pip install -qqq transformers>=4.39.0 mamba-ssm causal-conv1d>=1.2.0 accelerate bitsandbytes --progress-bar off
 !pip install flash-attn --no-build-isolation