pglo commited on
Commit
7834475
·
verified ·
1 Parent(s): eba0fcb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -5
README.md CHANGED
@@ -25,11 +25,15 @@ Zamba2-1.2B-Instruct is a hybrid model composed of state-space ([Mamba2](https:/
25
 
26
  ### Prerequisites
27
 
28
- To download Zamba2-1.2B, clone Zyphra's fork of transformers:
29
- 1. `git clone https://github.com/Zyphra/transformers_zamba2.git`
30
- 2. `cd transformers_zamba2`
31
- 3. Install the repository: `pip install -e .`
32
- 4. `pip install accelerate`
 
 
 
 
33
 
34
 
35
  You can run the model without using the optimized Mamba2 kernels, but it is **not** recommended as it will result in significantly higher latency and memory usage.
 
25
 
26
  ### Prerequisites
27
 
28
+ To use Zamba2-1.2B-instruct, install `transformers`:
29
+
30
+ `pip install transformers`
31
+
32
+ To install dependencies necessary to run Mamba2 kernels, install `mamba-ssm` from source (due to compatibility issues with PyTorch) as well as `causal-conv1d`:
33
+
34
+ 1. `git clone https://github.com/state-spaces/mamba.git`
35
+ 2. `cd mamba && git checkout v2.1.0 && pip install .`
36
+ 3. `pip install causal-conv1d`
37
 
38
 
39
  You can run the model without using the optimized Mamba2 kernels, but it is **not** recommended as it will result in significantly higher latency and memory usage.