stanfordnlp/imdb
Viewer • Updated • 100k • 171k • 1.27k
How to use Iamyulx/sentiment-bert-imdb with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Iamyulx/sentiment-bert-imdb", device_map="auto")Base model
google-bert/bert-base-uncased