How to use nreimers/MiniLM-L6-H384-uncased with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="nreimers/MiniLM-L6-H384-uncased")
# 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")
This is a 6 layer version of microsoft/MiniLM-L12-H384-uncased by keeping only every second layer.
