-
Notifications
You must be signed in to change notification settings - Fork 216
Rag example not working #1688
Description
I ran the rag example in intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/README.md and I got this error -

Steps to re-create -
-
!apt-get install -y ffmpeg
!apt-get install -y libgl1-mesa-glx libgl1-mesa-dev -
!pip3 install intel-extension-for-transformers
-
!git clone https://github.com/intel/intel-extension-for-transformers.git
-
%cd ./intel-extension-for-transformers/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/
!pip install -r requirements.txt
%cd ../../../../../../ -
%cd ./intel-extension-for-transformers/intel_extension_for_transformers/neural_chat/
!pip install -r requirements.txt
%cd ../../../ -
then I ran rag example -
from intel_extension_for_transformers.neural_chat import PipelineConfig
from intel_extension_for_transformers.neural_chat import build_chatbot
from intel_extension_for_transformers.neural_chat import plugins
plugins.retrieval.enable=True
plugins.retrieval.args["input_path"]="./docs/"
config = PipelineConfig(plugins=plugins)
chatbot = build_chatbot(config)
and got the error -