Update README.md

README.md

@@ -6,7 +6,7 @@ license: apache-2.0
 
 
 
- [![Python](https://img.shields.io/badge/Python-3.10%2B-blue)](https://www.python.org/)  [![PyTorch](https://img.shields.io/badge/PyTorch-2.4.1-blue)](https://pytorch.org/) 
+ [![Python](https://img.shields.io/badge/Python-3.10%2B-blue)](https://www.python.org/)  [![PyTorch](https://img.shields.io/badge/PyTorch-2.6.0-blue)](https://pytorch.org/) 
 
 # FLAME: Flow Enhanced Legendre Memory Models for General Time Series Forecasting
 
@@ -52,5 +52,3 @@ pip install mamba-ssm[causal-conv1d]
 pip install zuko
 ```
 
-For mamba-ssm, there exists a bug about triton. We fix it and release the code at [./flame/mamba/mamba_ssm](./flame/mamba/mamba_ssm), you can compile it from scratch or simply replace the mamba_ssm package under the default path of conda.
-