Skip to content

Commit 80388ca

Browse files
committed
Update download instruction for GitHub
1 parent 5fba6e0 commit 80388ca

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

README.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,21 @@ You can use LC-PLM to extract embeddings for amino acid residues and protein seq
1717
```bash
1818
pip install transformers mamba-ssm==2.2.2
1919
```
20+
### Clone this repo with pretrained model weights
2021

21-
### Download and run inference with the pretrained model
22+
We use [Git Large File Storage (LFS)](https://git-lfs.com/) to version the model weights. You can obtain the pretrained model and its related files simply by cloning this repo:
23+
```bash
24+
git clone https://github.com/amazon-science/LC-PLM.git
25+
```
26+
27+
### Run inference with the pretrained model
2228

2329
```python
2430
import torch
2531
from transformers import AutoTokenizer, AutoModelForMaskedLM
2632

2733
# Load the model and tokenizer
28-
model = AutoModelForMaskedLM.from_pretrained("amazon/LC-PLM", trust_remote_code=True)
34+
model = AutoModelForMaskedLM.from_pretrained("./LC-PLM", trust_remote_code=True)
2935
tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
3036

3137
# Input a protein sequence:

0 commit comments

Comments
 (0)