File size: 345 Bytes
10e5ef2 1452df5 10e5ef2 |
1 2 3 4 5 6 7 |
# Custom RoBERTa (safetensors)
Load with:
```python
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()
|