BERT Tiny Prompts
This model is a copy of stsb-bert-tiny-safetensors with prompts added for testing. It's intended for use with unit tests.
- Downloads last month
- 1,363
This model is a copy of stsb-bert-tiny-safetensors with prompts added for testing. It's intended for use with unit tests.
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("NeuML/bert-tiny-prompts") model = AutoModel.from_pretrained("NeuML/bert-tiny-prompts", device_map="auto")