stanfordnlp/imdb
Viewer • Updated • 100k • 182k • 1.3k
How to use jqop/distilledBERT-fintuned_with_imdb_dataset with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="jqop/distilledBERT-fintuned_with_imdb_dataset") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("jqop/distilledBERT-fintuned_with_imdb_dataset", device_map="auto")