summaryrefslogtreecommitdiff
path: root/configs/default.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'configs/default.yaml')
-rw-r--r--configs/default.yaml21
1 files changed, 21 insertions, 0 deletions
diff --git a/configs/default.yaml b/configs/default.yaml
new file mode 100644
index 0000000..cb76d34
--- /dev/null
+++ b/configs/default.yaml
@@ -0,0 +1,21 @@
+hopfield:
+ beta: 1.0
+ max_iter: 5
+ 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"