How to use explosion-testing/bert-test with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="explosion-testing/bert-test")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("explosion-testing/bert-test") model = AutoModelForMaskedLM.from_pretrained("explosion-testing/bert-test", device_map="auto")
Daniël de Kok +1: SFconvertbot Verified signature- Adding `safetensors` variant of this model (6720e3a3ab4e1baae714a5094f8391255587600e) Co-authored-by: Safetensors convertbot <SFconvertbot@users.noreply.huggingface.co>a1263d9