Skip to content

Commit

Permalink
back to MiniLM
Browse files Browse the repository at this point in the history
  • Loading branch information
Ludovico Mori committed Nov 3, 2023
1 parent ed98c24 commit 1c2e7a2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion config/dev-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ utils:
HuggingFaceEmbeddings:
class: HuggingFaceEmbeddings
kwargs:
model_name: sentence-transformers/multi-qa-mpnet-base-dot-v1 #sentence-transformers/all-MiniLM-L6-v2 #sentence-transformers/multi-qa-mpnet-base-dot-v1 #sentence-transformers/all-mpnet-base-v2 are better but currently not working
model_name: sentence-transformers/all-MiniLM-L6-v2 #sentence-transformers/multi-qa-mpnet-base-dot-v1 #sentence-transformers/all-mpnet-base-v2 are better but currently not working
model_kwargs:
device: 'cpu'
encode_kwargs:
Expand Down

0 comments on commit 1c2e7a2

Please sign in to comment.