Edit model card

ONNX version of huggingface/CodeBERTa-language-id

This model is conversion of huggingface/CodeBERTa-language-id to ONNX. The model was converted to ONNX using the 🤗 Optimum library.

Model Architecture

Base Model: CodeBERTa, a variant of the RoBERTa model trained specifically for programming languages.

Modifications: No changes except for the conversion.

Usage

Optimum

Loading the model requires the 🤗 Optimum library installed.

from optimum.onnxruntime import ORTModelForSequenceClassification
from transformers import AutoTokenizer, pipeline


tokenizer = AutoTokenizer.from_pretrained("laiyer/CodeBERTa-language-id")
model = ORTModelForSequenceClassification.from_pretrained("laiyer/CodeBERTa-language-id")
classifier = pipeline(
    task="text-classification",
    model=model,
    tokenizer=tokenizer,
)

print(classifier("""
def f(x):
    return x**2
"""))

LLM Guard

Code scanner

Community

Join our Slack to give us feedback, connect with the maintainers and fellow users, ask questions, or engage in discussions about LLM security!

Downloads last month
10
Inference Examples
Inference API (serverless) has been turned off for this model.

Model tree for protectai/CodeBERTa-language-id-onnx

Quantized
(1)
this model

Dataset used to train protectai/CodeBERTa-language-id-onnx

Collection including protectai/CodeBERTa-language-id-onnx