added cache dir

AllenNeuralDynamics · Nov 21, 2024 · f71065b · f71065b
1 parent 15f586f
commit f71065b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -14,6 +14,7 @@ umap_visualization.py
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
+huggingface_cache/
 *.py[cod]
 *$py.class
 

diff --git a/src/metadata_chatbot/agents/docdb_retriever.py b/src/metadata_chatbot/agents/docdb_retriever.py
@@ -17,7 +17,8 @@
 hf = HuggingFaceEmbeddings(
     model_name=model_name,
     model_kwargs=model_kwargs,
-    encode_kwargs=encode_kwargs
+    encode_kwargs=encode_kwargs,
+    cache_folder="/scratch/huggingface_cache"
 )
 
 BEDROCK_CLIENT = boto3.client(