How to use alfonsusrr/BugBERTa-C with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="alfonsusrr/BugBERTa-C")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("alfonsusrr/BugBERTa-C") model = AutoModelForSequenceClassification.from_pretrained("alfonsusrr/BugBERTa-C", device_map="auto")
This model is trained on base model microsoft/codebert-base added with classification head and trained with DebugBench dataset for bug classification purpose.
microsoft/codebert-base
DebugBench
Files info
