AI4Protein

/

ProPrime_650M

like 3

Follow

Venus 58

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("fill-mask", model="AI4Protein/ProPrime_650M", trust_remote_code=True)

# Load model directly
from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained("AI4Protein/ProPrime_650M", trust_remote_code=True, dtype="auto")