diff --git a/config/dev-config.yaml b/config/dev-config.yaml index 9e0d094..8ddca9c 100644 --- a/config/dev-config.yaml +++ b/config/dev-config.yaml @@ -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: