summaryrefslogtreecommitdiff
path: root/configs/hotpotqa.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'configs/hotpotqa.yaml')
-rw-r--r--configs/hotpotqa.yaml22
1 files changed, 22 insertions, 0 deletions
diff --git a/configs/hotpotqa.yaml b/configs/hotpotqa.yaml
new file mode 100644
index 0000000..99ea9d7
--- /dev/null
+++ b/configs/hotpotqa.yaml
@@ -0,0 +1,22 @@
+hopfield:
+ beta: 2.0
+ max_iter: 8
+ conv_threshold: 1.0e-4
+ top_k: 5
+
+memory:
+ embedding_dim: 768
+ normalize: true
+
+encoder:
+ model_name: "facebook/contriever-msmarco"
+ max_length: 512
+ batch_size: 64
+
+generator:
+ model_name: "meta-llama/Llama-3.1-8B-Instruct"
+ max_new_tokens: 128
+ temperature: 0.0
+
+retriever_type: "hopfield"
+dataset: "hotpotqa"