ONNX-converted models
Collection
Models converted to ONNX for faster CPU inference on LLM Guard. • 20 items • Updated • 2
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("protectai/vishnun-codenlbert-tiny-onnx")
model = AutoModelForSequenceClassification.from_pretrained("protectai/vishnun-codenlbert-tiny-onnx", device_map="auto")Quick Links
THIS PROJECT HAS BEEN ARCHIVED.
This project and its associated code on GitHub are no longer under active development or maintained.
ONNX version of vishnun/codenlbert-tiny
This model is a conversion of vishnun/codenlbert-tiny to ONNX format using the 🤗 Optimum library.
- Downloads last month
- 12
Model tree for protectai/vishnun-codenlbert-tiny-onnx
Base model
vishnun/codenlbert-tiny
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="protectai/vishnun-codenlbert-tiny-onnx")