diff --git a/.gitignore b/.gitignore index cabac1d..fa161a2 100644 --- 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 index 283378e..a1f75f7 100644 --- 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(