fix init issue of langchain chroma (#417)

intel · Oct 9, 2023 · fdefe27 · fdefe27
1 parent f5d34ed
commit fdefe27
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/.github/workflows/script/chatbot/start_ft_mpt-7b-chat_mpi.sh b/.github/workflows/script/chatbot/start_ft_mpt-7b-chat_mpi.sh
@@ -3,10 +3,10 @@ hname=$(hostname -s)
 python3 workflows/chatbot/fine_tuning/instruction_tuning_pipeline/finetune_clm.py \
     --model_name_or_path mosaicml/mpt-7b-chat \
     --train_file .github/workflows/sample_data/alpaca_data_sample_45.json \
-    --bf16 True \
+    --bf16 False \
     --output_dir ./mpt_peft_finetuned_model \
     --num_train_epochs 1 \
-    --max_steps 3
+    --max_steps 3 \
     --per_device_train_batch_size 4 \
     --per_device_eval_batch_size 4 \
     --gradient_accumulation_steps 1 \

diff --git a/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/indexing/indexing.py b/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/indexing/indexing.py
@@ -21,7 +21,6 @@
 from langchain.vectorstores.chroma import Chroma
 from langchain.docstore.document import Document
 from langchain.embeddings import HuggingFaceInstructEmbeddings
-from langchain.vectorstores import Chroma
 from haystack.schema import Document as SDocument
 from .context_utils import load_unstructured_data, laod_structured_data, get_chuck_data
 

diff --git a/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/retrieval_chroma.py b/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/retrieval_chroma.py
@@ -17,7 +17,7 @@
 
 """The wrapper for Chroma retriever based on langchain"""
 
-from langchain.vectorstores import Chroma
+from langchain.vectorstores.chroma import Chroma
 
 class ChromaRetriever():
     """Retrieve the document database with Chroma database using dense retrieval."""

diff --git a/intel_extension_for_transformers/neural_chat/requirements.txt b/intel_extension_for_transformers/neural_chat/requirements.txt
@@ -14,7 +14,7 @@ evaluate
 pydub
 python-multipart
 PyPDF2
-langchain==0.0.308
+langchain
 python-docx
 scikit-learn
 farm-haystack