Upload training_args.bin 6b1ca4a
Joana Palés Huix commited on
How to use Peltarion/dnabert-minilm-mini with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForPreTraining
tokenizer = AutoTokenizer.from_pretrained("Peltarion/dnabert-minilm-mini")
model = AutoModelForPreTraining.from_pretrained("Peltarion/dnabert-minilm-mini", device_map="auto")