legacy-datasets/wikipedia
Updated • 160k • 661
How to use GeneZC/sparsebert-base-minilm-99S with Transformers:
# Load model directly
from transformers import AutoTokenizer, SparseBertMiniLM
tokenizer = AutoTokenizer.from_pretrained("GeneZC/sparsebert-base-minilm-99S")
model = SparseBertMiniLM.from_pretrained("GeneZC/sparsebert-base-minilm-99S", device_map="auto")
# Load model directly from transformers import AutoTokenizer, SparseBertMiniLM tokenizer = AutoTokenizer.from_pretrained("GeneZC/sparsebert-base-minilm-99S") model = SparseBertMiniLM.from_pretrained("GeneZC/sparsebert-base-minilm-99S", device_map="auto")