SCAI-BIO

/

bio-gottbert-base

Like 4

Follow

AI & Data Science, Department of Bioinformatics, Fraunhofer Institute for Algorithms and Scientific Computing (SCAI) 3

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

pipe = pipeline("fill-mask", model="SCAI-BIO/bio-gottbert-base")

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM

tokenizer = AutoTokenizer.from_pretrained("SCAI-BIO/bio-gottbert-base")
model = AutoModelForMaskedLM.from_pretrained("SCAI-BIO/bio-gottbert-base", device_map="auto")