Instructions to use unitary/toxic-bert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use unitary/toxic-bert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="unitary/toxic-bert")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("unitary/toxic-bert") model = AutoModelForSequenceClassification.from_pretrained("unitary/toxic-bert") - Inference
- Notebooks
- Google Colab
- Kaggle
Upload tokenizer.json
#9
by ada-closure - opened
Generated with:
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("unitary/toxic-bert")
assert tokenizer.is_fast
tokenizer.save_pretrained("...")
This was generated for use in Bumblebee which relies on the Rust based tokenizer. For more information please refer to this link:
https://github.com/elixir-nx/bumblebee?tab=readme-ov-file#tokenizer-support