How to use SeprotHub/Model-Classification-ProTrek-35M with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("ProTrekHub/Protein_Encoder_35M") model = PeftModel.from_pretrained(base_model, "SeprotHub/Model-Classification-ProTrek-35M")
797ccde
1
2
3
4
{ "training_task_type": "Protein-level Classification", "training_data_type": "AA" }