# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("nigelhartm/PlantBERT")
model = AutoModelForMaskedLM.from_pretrained("nigelhartm/PlantBERT", device_map="auto")Quick Links
#PlantBERT
Pre-trained BERT model and BPE tokenizer utilizing only plant genome data. More information at GitHub: https://github.com/nigelhartm/PlantBERT
- Downloads last month
- 12
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="nigelhartm/PlantBERT")