Skip to content

Commit 7643232

Browse files
cheungdavenUbuntu
andauthored
config bug fix (#96)
Co-authored-by: Ubuntu <[email protected]>
1 parent ff09ce3 commit 7643232

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/agrag/agrag.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def __init__(
126126
self.model_ids = model_ids
127127

128128
self.config = config_file or self._load_preset()
129-
self.args = Arguments(self.config) if not self.args else self.args
129+
self.args = Arguments(self.config) if self.config else self.args
130130

131131
# will short-circuit to provided data_dir if config value also provided
132132
self.data_dir = data_dir or self.args.data_dir

src/agrag/configs/presets/medium_quality_config.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ embedding:
1313
embedding_model_platform: bedrock
1414
embedding_model_platform_args:
1515
bedrock_aws_region: us-west-2
16+
hf_tokenizer_params:
17+
truncation: true
18+
padding: true
1619
normalize_embeddings: false
17-
hf_tokenizer_params:
18-
truncation: true
19-
padding: true
2020
normalization_params: {'p': 2, 'dim': 1, 'eps': 1e-12}
2121
embedding_batch_size: 64
2222

0 commit comments

Comments
 (0)