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 +0 -1
config.json
CHANGED
|
@@ -19,7 +19,6 @@
|
|
| 19 |
"output_past": true,
|
| 20 |
"pad_token_id": 1,
|
| 21 |
"position_embedding_type": "absolute",
|
| 22 |
-
"problem_type": "multi_task_classification",
|
| 23 |
"torch_dtype": "float32",
|
| 24 |
"transformers_version": "4.47.0",
|
| 25 |
"type_vocab_size": 1,
|
|
|
|
| 19 |
"output_past": true,
|
| 20 |
"pad_token_id": 1,
|
| 21 |
"position_embedding_type": "absolute",
|
|
|
|
| 22 |
"torch_dtype": "float32",
|
| 23 |
"transformers_version": "4.47.0",
|
| 24 |
"type_vocab_size": 1,
|