Skip to content

Commit

Permalink
config bug fix (#96)
Browse files Browse the repository at this point in the history
Co-authored-by: Ubuntu <ubuntu@ip-172-31-3-59.us-west-2.compute.internal>
  • Loading branch information
cheungdaven and Ubuntu authored Dec 18, 2024
1 parent ff09ce3 commit 7643232
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion src/agrag/agrag.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ def __init__(
self.model_ids = model_ids

self.config = config_file or self._load_preset()
self.args = Arguments(self.config) if not self.args else self.args
self.args = Arguments(self.config) if self.config else self.args

# will short-circuit to provided data_dir if config value also provided
self.data_dir = data_dir or self.args.data_dir
Expand Down
6 changes: 3 additions & 3 deletions src/agrag/configs/presets/medium_quality_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,10 @@ embedding:
embedding_model_platform: bedrock
embedding_model_platform_args:
bedrock_aws_region: us-west-2
hf_tokenizer_params:
truncation: true
padding: true
normalize_embeddings: false
hf_tokenizer_params:
truncation: true
padding: true
normalization_params: {'p': 2, 'dim': 1, 'eps': 1e-12}
embedding_batch_size: 64

Expand Down

0 comments on commit 7643232

Please sign in to comment.