Duplicated from Shubhamai/tiny-bert-for-token-classification
How to use moosemandada/tiny-ner-probe with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="moosemandada/tiny-ner-probe")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("moosemandada/tiny-ner-probe") model = AutoModelForTokenClassification.from_pretrained("moosemandada/tiny-ner-probe", device_map="auto")
https://huggingface.co/moosemandada/tiny-ner-probe/resolve/main/special_tokens_map.json
hf download hf://moosemandada/tiny-ner-probe/special_tokens_map.json
curl -L -o special_tokens_map.json https://huggingface.co/moosemandada/tiny-ner-probe/resolve/main/special_tokens_map.json
