Instructions to use CyberPeace-Institute/Cybersecurity-Knowledge-Graph with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use CyberPeace-Institute/Cybersecurity-Knowledge-Graph with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="CyberPeace-Institute/Cybersecurity-Knowledge-Graph", trust_remote_code=True)# Load model directly from transformers import AutoModelForTokenClassification model = AutoModelForTokenClassification.from_pretrained("CyberPeace-Institute/Cybersecurity-Knowledge-Graph", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
21c1379
1
Parent(s): 400226a
Upload model
Browse files- config.json +4 -0
config.json
CHANGED
|
@@ -2,6 +2,10 @@
|
|
| 2 |
"architectures": [
|
| 3 |
"CybersecurityKnowledgeGraphModel"
|
| 4 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
"event_argument_model_path": "cybersecurity_knowledge_graph/argument_model_state_dict.pth",
|
| 6 |
"event_nugget_model_path": "cybersecurity_knowledge_graph/nugget_model_state_dict.pth",
|
| 7 |
"event_realis_model_path": "cybersecurity_knowledge_graph/realis_model_state_dict.pth",
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"CybersecurityKnowledgeGraphModel"
|
| 4 |
],
|
| 5 |
+
"auto_map": {
|
| 6 |
+
"AutoConfig": "configuration.CybersecurityKnowledgeGraphConfig",
|
| 7 |
+
"AutoModelForTokenClassification": "model.CybersecurityKnowledgeGraphModel"
|
| 8 |
+
},
|
| 9 |
"event_argument_model_path": "cybersecurity_knowledge_graph/argument_model_state_dict.pth",
|
| 10 |
"event_nugget_model_path": "cybersecurity_knowledge_graph/nugget_model_state_dict.pth",
|
| 11 |
"event_realis_model_path": "cybersecurity_knowledge_graph/realis_model_state_dict.pth",
|