LLM: fix langchain windows failure (#8417)

This commit is contained in:
binbin Deng 2023-06-30 09:59:10 +08:00 committed by GitHub
parent 6251ad8934
commit 146662bc0d
3 changed files with 19 additions and 3 deletions

View file

@ -64,6 +64,10 @@ jobs:
- name: Run LLM unittests - name: Run LLM unittests
shell: bash shell: bash
run: | run: |
pip install pytest pydantic
pip install -U langchain==0.0.184
pip install -U chromadb==0.3.25
pip install -U typing_extensions==4.5.0
bash python/llm/test/run-llm-windows-tests.sh bash python/llm/test/run-llm-windows-tests.sh
env: env:
ANALYTICS_ZOO_ROOT: ${{ github.workspace }} ANALYTICS_ZOO_ROOT: ${{ github.workspace }}

View file

@ -49,7 +49,7 @@ class Test_Models_Basics(TestCase):
llm = BigdlLLM( llm = BigdlLLM(
model_path=self.llama_model_path, model_path=self.llama_model_path,
max_tokens=32, max_tokens=32,
n_threads=22) n_threads=2)
question = "What is AI?" question = "What is AI?"
result = llm(question) result = llm(question)
@ -58,7 +58,7 @@ class Test_Models_Basics(TestCase):
model_path=self.gptneox_model_path, model_path=self.gptneox_model_path,
model_family="gptneox", model_family="gptneox",
max_tokens=32, max_tokens=32,
n_threads=22) n_threads=2)
question = "What is AI?" question = "What is AI?"
result = llm(question) result = llm(question)
@ -67,7 +67,7 @@ class Test_Models_Basics(TestCase):
model_path=self.bloom_model_path, model_path=self.bloom_model_path,
model_family="bloom", model_family="bloom",
max_tokens=32, max_tokens=32,
n_threads=22) n_threads=2)
question = "What is AI?" question = "What is AI?"
result = llm(question) result = llm(question)

View file

@ -44,3 +44,15 @@ time=$((now-start))
echo "Bigdl-llm inference test finished" echo "Bigdl-llm inference test finished"
echo "Time used:$time seconds" echo "Time used:$time seconds"
echo "# Start testing langchain"
start=$(date "+%s")
python -m pytest -s ${ANALYTICS_ZOO_ROOT}/python/llm/test/langchain/test_langchain.py -k 'test_langchain_llm_bloom'
now=$(date "+%s")
time=$((now-start))
echo "Bigdl-llm langchain test finished"
echo "Time used:$time seconds"