SkywalkerLu commited on
Commit
7285f3b
·
verified ·
1 Parent(s): aed7b7f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -43,7 +43,7 @@ from transformers import AutoModel, AutoTokenizer
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
44
 
45
  # Load model (replace with your model id if different)
46
- model_id = "SkywalkerLu/TransHLA2.0"
47
  model = AutoModel.from_pretrained(model_id, trust_remote_code=True).to(device).eval()
48
 
49
  # Load tokenizer used in training (ESM2 650M)
 
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
44
 
45
  # Load model (replace with your model id if different)
46
+ model_id = "SkywalkerLu/TransHLA2.0-BIND"
47
  model = AutoModel.from_pretrained(model_id, trust_remote_code=True).to(device).eval()
48
 
49
  # Load tokenizer used in training (ESM2 650M)