Instructions to use bziemba/SecureBERT-MultiTask-CVSS-CWE with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use bziemba/SecureBERT-MultiTask-CVSS-CWE with Transformers:
# Load model directly from transformers import AutoTokenizer, SecureBERTMultiHead tokenizer = AutoTokenizer.from_pretrained("bziemba/SecureBERT-MultiTask-CVSS-CWE") model = SecureBERTMultiHead.from_pretrained("bziemba/SecureBERT-MultiTask-CVSS-CWE") - Notebooks
- Google Colab
- Kaggle
| { | |
| "architectures": [ | |
| "SecureBERTMultiHead" | |
| ], | |
| "model_type": "roberta", | |
| "base_model": "cisco-ai/SecureBERT2.0-biencoder", | |
| "custom_heads": [ | |
| "cvss", | |
| "cwe" | |
| ] | |
| } |