Haixx commited on
Commit
6b07cce
1 Parent(s): 81aa60a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -2
README.md CHANGED
@@ -7,5 +7,21 @@ datasets:
7
 
8
  # DARA: Decomposition-Alignment-Reasoning Autonomous Language Agent for Question Answering over Knowledge Graphs
9
 
10
- <!-- Provide a quick summary of what the model is/does. -->
11
- This model is a fine-tuned semantic parsing LLM agent for KGQA. Please check the repository https://github.com/UKPLab/acl2024-DARA for details.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
  # DARA: Decomposition-Alignment-Reasoning Autonomous Language Agent for Question Answering over Knowledge Graphs
9
 
10
+ ## Model Information
11
+ This model is a fine-tuned semantic parsing LLM agent for KGQA. We fine-tune the llama-2-13B on our curated reasoning trajectory https://huggingface.co/datasets/UKPLab/dara.
12
+
13
+ Model Usage
14
+ from transformers import AutoModelForCausalLM
15
+
16
+ model = AutoModelForCausalLM.from_pretrained(
17
+ "UKPLab/dara-llama-2-13b",
18
+ torch_dtype=torch.float16,
19
+ device_map="auto",
20
+ cache_dir = "cache"
21
+ )
22
+ For more information, please check the repository https://github.com/UKPLab/acl2024-DARA
23
+
24
+ Hyperparameters
25
+ Learning rate: 2e-5
26
+ Batch size: 4
27
+ Training epochs: 10