sarahalamdari commited on
Commit
aaeb4fa
·
verified ·
1 Parent(s): e252ccb

Add installation instructions

Browse files
Files changed (1) hide show
  1. README.md +36 -1
README.md CHANGED
@@ -45,7 +45,42 @@ This model should not be used to generate anything that is not a protein sequenc
45
 
46
  ## How to Get Started with the Model
47
 
48
- Sample protein generation code:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
49
 
50
  ```py
51
 
 
45
 
46
  ## How to Get Started with the Model
47
 
48
+ The simplest way to use these models and datasets is via the HuggingFace interface. You will need PyTorch, mamba=ssm, causal-conv1d, and flash-attn.
49
+
50
+ **Requirements**:
51
+ * PyTorch: 2.7.1
52
+ * CUDA 12.8 and above
53
+
54
+ We recommend using [uv](https://docs.astral.sh/uv/getting-started/installation/#standalone-installer) and creating a clean environment.
55
+
56
+
57
+ ```bash
58
+ uv venv dayhoff
59
+ source dayhoff/bin/activate
60
+ ```
61
+
62
+ In that new environment, install PyTorch 2.7.1.
63
+ ```bash
64
+ uv pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu128
65
+ ```
66
+
67
+ Now, we need to install mamba-ssm, flash-attn, causal-conv1d, and their prerequisites.
68
+
69
+ ```bash
70
+ uv pip install wheel packaging
71
+ uv pip install --no-build-isolation flash-attn causal-conv1d mamba-ssm
72
+ ```
73
+
74
+ To import from HuggingFace, you will need to install these versions:
75
+
76
+ ```bash
77
+ uv pip install datasets==3.2.0 #for HF datasets
78
+ uv pip install transformers==4.51.0
79
+ uv pip install huggingface_hub~=0.34.4
80
+ ```
81
+
82
+ **Sample protein generation code:**
83
+
84
 
85
  ```py
86