Instructions to use mahdin70/CodeBERT-PrimeVul-BigVul with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mahdin70/CodeBERT-PrimeVul-BigVul with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mahdin70/CodeBERT-PrimeVul-BigVul", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("mahdin70/CodeBERT-PrimeVul-BigVul", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload 2 files
Browse files- model.safetensors +3 -0
- training_args.bin +3 -0
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b058f30a6ce46510650a3a0f545eca2ccbaef74945c411787251b393a104848
|
| 3 |
+
size 499027652
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:870d028853c45fe3e244895e4bae129ce4dfc5fb6ba9388afb1cfd0dc3fc2e6b
|
| 3 |
+
size 5240
|