ESM-PTM: ESM-2 for Predicting PTM
Collection
This is a collection of LoRA and QLoRA finetuned ESM-2 models and datasets for predicting post translational modification sites on proteins. • 6 items • Updated
How to use AmelieSchreiber/esm2_t30_150M_ptm_qlora_2100K with PEFT:
from peft import PeftModel
from transformers import AutoModelForTokenClassification
base_model = AutoModelForTokenClassification.from_pretrained("facebook/esm2_t30_150M_UR50D")
model = PeftModel.from_pretrained(base_model, "AmelieSchreiber/esm2_t30_150M_ptm_qlora_2100K")