amp-generation

/

AMP-ProGen2

like 1

Follow

Antimicrobial Peptide Generation 3

# Gated model: Login with a HF token with gated access permission
hf auth login

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

pipe = pipeline("text-generation", model="amp-generation/AMP-ProGen2")

# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("amp-generation/AMP-ProGen2", dtype="auto")