Text Classification
Transformers
Safetensors
multi_task_unixcoder
feature-extraction
Code
Vulnerability
Detection
custom_code
Instructions to use mahdin70/UnixCoder-Primevul-BigVul with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mahdin70/UnixCoder-Primevul-BigVul with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mahdin70/UnixCoder-Primevul-BigVul", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("mahdin70/UnixCoder-Primevul-BigVul", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -24,7 +24,7 @@
|
|
| 24 |
"type_vocab_size": 1,
|
| 25 |
"use_cache": true,
|
| 26 |
"vocab_size": 50265,
|
| 27 |
-
"num_cwe_classes":
|
| 28 |
"auto_map": {
|
| 29 |
"AutoConfig": "modeling_multi_task_unixcoder.MultiTaskUnixCoderConfig",
|
| 30 |
"AutoModel": "modeling_multi_task_unixcoder.MultiTaskUnixCoder"
|
|
|
|
| 24 |
"type_vocab_size": 1,
|
| 25 |
"use_cache": true,
|
| 26 |
"vocab_size": 50265,
|
| 27 |
+
"num_cwe_classes": 134,
|
| 28 |
"auto_map": {
|
| 29 |
"AutoConfig": "modeling_multi_task_unixcoder.MultiTaskUnixCoderConfig",
|
| 30 |
"AutoModel": "modeling_multi_task_unixcoder.MultiTaskUnixCoder"
|