Update README.md
Browse files
README.md
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
pipeline_tag: feature-extraction
|
| 4 |
-
model_name:
|
| 5 |
---
|
| 6 |
|
| 7 |
# IDP-ESM2-8M
|
|
@@ -20,7 +20,7 @@ from transformers import AutoTokenizer, AutoModel
|
|
| 20 |
import torch
|
| 21 |
|
| 22 |
# --- Config ---
|
| 23 |
-
model_name = "
|
| 24 |
|
| 25 |
# --- Load model and tokenizer ---
|
| 26 |
tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
|
|
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
pipeline_tag: feature-extraction
|
| 4 |
+
model_name: InstaDeepAI/IDP-ESM2-150M
|
| 5 |
---
|
| 6 |
|
| 7 |
# IDP-ESM2-8M
|
|
|
|
| 20 |
import torch
|
| 21 |
|
| 22 |
# --- Config ---
|
| 23 |
+
model_name = "InstaDeepAI/IDP-ESM2-150M"
|
| 24 |
|
| 25 |
# --- Load model and tokenizer ---
|
| 26 |
tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
|