legacy-datasets/wikipedia
Updated • 155k • 660
How to use GeneZC/sparsebert-base with Transformers:
# Load model directly
from transformers import AutoTokenizer, SparseBertHidden
tokenizer = AutoTokenizer.from_pretrained("GeneZC/sparsebert-base")
model = SparseBertHidden.from_pretrained("GeneZC/sparsebert-base", device_map="auto")