eunJ's picture
initial push (safetensors + custom head)
10e5ef2 verified

Custom RoBERTa (safetensors)

Load with:

from transformers import AutoTokenizer, AutoModelForSequenceClassification
tok = AutoTokenizer.from_pretrained("eunJ/codebert_vulnerability_detector_multi")
model = AutoModelForSequenceClassification.from_pretrained("eunJ/codebert_vulnerability_detector_multi", trust_remote_code=True).eval()