legacy-datasets/wikipedia
Updated • 136k • 663
How to use GeneZC/bert-large-minilm-4L-384H with Transformers:
# 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")