# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("nreimers/BERT-Tiny_L-2_H-128_A-2")
model = AutoModel.from_pretrained("nreimers/BERT-Tiny_L-2_H-128_A-2")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
This is the BERT-Medium model from Google: https://github.com/google-research/bert#bert. A BERT model with 2 layers, 128 hidden unit size, and 2 attention heads.
- Downloads last month
- 15,203
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="nreimers/BERT-Tiny_L-2_H-128_A-2")