ESMBind (ESMB) for Protein Binding Sites
Collection
ESMBind (or ESMB) are LoRA finetuned ESM-2 models for predicting general binding sites and active sites of proteins based on sequence alone. • 14 items • Updated • 5
How to use AmelieSchreiber/esm2_t12_35M_lora_binding_sites_cp2 with PEFT:
from peft import PeftModel
from transformers import AutoModelForTokenClassification
base_model = AutoModelForTokenClassification.from_pretrained("facebook/esm2_t12_35M_UR50D")
model = PeftModel.from_pretrained(base_model, "AmelieSchreiber/esm2_t12_35M_lora_binding_sites_cp2")
from peft import PeftModel from transformers import AutoModelForTokenClassification base_model = AutoModelForTokenClassification.from_pretrained("facebook/esm2_t12_35M_UR50D") model = PeftModel.from_pretrained(base_model, "AmelieSchreiber/esm2_t12_35M_lora_binding_sites_cp2")