hugging2021 commited on
Commit
1d98a7e
·
verified ·
1 Parent(s): 3c9ce69

Update concat_vector_store_정리된.py

Browse files
Files changed (1) hide show
  1. concat_vector_store_정리된.py +1 -1
concat_vector_store_정리된.py CHANGED
@@ -10,7 +10,7 @@ FOLDER = "cleaned_pdfs" # Folder containing the cleaned PDFs
10
  VECTOR_STORE_PATH = "vector_db"
11
 
12
  # 1. Load the embedding model
13
- def get_embeddings(model_name="intfloat/multilingual-e5-large-instruct", device="cuda"):
14
  return E5Embeddings(
15
  model_name=model_name,
16
  model_kwargs={'device': device},
 
10
  VECTOR_STORE_PATH = "vector_db"
11
 
12
  # 1. Load the embedding model
13
+ def get_embeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", device="cuda"):
14
  return E5Embeddings(
15
  model_name=model_name,
16
  model_kwargs={'device': device},