diff options
Diffstat (limited to 'configs/default.yaml')
| -rw-r--r-- | configs/default.yaml | 21 |
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" |
