mgbam commited on
Commit
9920af3
·
verified ·
1 Parent(s): c50fa3d

Update retrieval.py

Browse files
Files changed (1) hide show
  1. retrieval.py +2 -2
retrieval.py CHANGED
@@ -79,10 +79,10 @@ class EmbedFunction:
79
  EMBED_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
80
  embed_function = EmbedFunction(EMBED_MODEL_NAME)
81
 
82
- # Use in-memory storage to avoid persistent storage issues.
83
  client = chromadb.Client(
84
  settings=Settings(
85
- persist_directory=":memory:",
86
  anonymized_telemetry=False
87
  )
88
  )
 
79
  EMBED_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
80
  embed_function = EmbedFunction(EMBED_MODEL_NAME)
81
 
82
+ # Use ephemeral (in-memory) storage to avoid persistent storage issues.
83
  client = chromadb.Client(
84
  settings=Settings(
85
+ in_memory=True,
86
  anonymized_telemetry=False
87
  )
88
  )