# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("nreimers/MiniLM-L6-H384-uncased")
model = AutoModel.from_pretrained("nreimers/MiniLM-L6-H384-uncased", device_map="auto")Quick Links
MiniLM: 6 Layer Version
This is a 6 layer version of microsoft/MiniLM-L12-H384-uncased by keeping only every second layer.
- Downloads last month
- 9,423
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="nreimers/MiniLM-L6-H384-uncased")