legacy-datasets/wikipedia
Updated โข 46.6k โข 688
How to use GeneZC/bert-large-minilm-4L-384H with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, BertMiniLM
tokenizer = AutoTokenizer.from_pretrained("GeneZC/bert-large-minilm-4L-384H")
model = BertMiniLM.from_pretrained("GeneZC/bert-large-minilm-4L-384H", device_map="auto")