# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("nreimers/MiniLM-L3-H384-uncased")
model = AutoModel.from_pretrained("nreimers/MiniLM-L3-H384-uncased", device_map="auto")Quick Links
MiniLM: 3 Layer Version
This is a 3 layer version of microsoft/MiniLM-L12-H384-uncased by keeping only the layer [3, 7, 11].
- Downloads last month
- 55,040
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="nreimers/MiniLM-L3-H384-uncased")