legacy-datasets/wikipedia
Updated • 160k • 661
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")