Load each model using:
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("gagneurlab/SpeciesLM", revision = "<<choose model type>>")
model = AutoModelForMaskedLM.from_pretrained("gagneurlab/SpeciesLM", revision = "<<choose model type>>")
Model type:
downstream_species_lmdownstream_agnostic_lmupstream_species_lmupstream_agnostic_lm1 commits
1 commits
Load each model using:
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("gagneurlab/SpeciesLM", revision = "<<choose model type>>")
model = AutoModelForMaskedLM.from_pretrained("gagneurlab/SpeciesLM", revision = "<<choose model type>>")
Model type:
downstream_species_lmdownstream_agnostic_lmupstream_species_lmupstream_agnostic_lm1 commits
1 commits