Instructions to use mahdin70/GraphCodeBERT-VulnCWE with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mahdin70/GraphCodeBERT-VulnCWE with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="mahdin70/GraphCodeBERT-VulnCWE", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("mahdin70/GraphCodeBERT-VulnCWE", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload 3 files
Browse files- config.json +9 -0
- model.safetensors +3 -0
- training_args.bin +3 -0
config.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"MultiTaskGraphCodeBERT"
|
| 4 |
+
],
|
| 5 |
+
"model_type": "multi_task_graphcodebert",
|
| 6 |
+
"num_cwe_classes": 12,
|
| 7 |
+
"torch_dtype": "float32",
|
| 8 |
+
"transformers_version": "4.51.3"
|
| 9 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ed7cf5d45cae3825c27d0882253484560560fe75eca07dba56257a1a7e78be6
|
| 3 |
+
size 498649304
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:297f0f3bbf4f7876bb3e7be416570272ad94c503d8f8ad74bff558615e0f4f13
|
| 3 |
+
size 5240
|