From 173014c219bc68177b445a250e49a2b820b1479a Mon Sep 17 00:00:00 2001 From: Quentin Anthony Date: Fri, 23 Aug 2024 09:53:38 -0700 Subject: [PATCH] add mamba --- README.md | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/README.md b/README.md index 6fccf5f79..28dfa6a31 100644 --- a/README.md +++ b/README.md @@ -41,6 +41,15 @@ cd apex pip install -r requirements.txt pip install -v --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./ ``` + +## Install mamba +``` +git clone https://github.com/state-spaces/mamba.git +cd mamba +pip install causal-conv1d>=1.4.0 +python setup.py install +``` + ## gpt-neox ``` git clone https://github.com/EleutherAI/gpt-neox.git