pip install neural_speed from wheel (#1184)

Signed-off-by: lvliang-intel <[email protected]>
intel · Jan 24, 2024 · 0f31c4b · 0f31c4b
1 parent 2adda8c
commit 0f31c4b
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/script/unitTest/run_unit_test_neuralchat.sh b/.github/workflows/script/unitTest/run_unit_test_neuralchat.sh
@@ -86,10 +86,6 @@ function main() {
     apt-get install libsm6 libxext6 -y
     wget http://nz2.archive.ubuntu.com/ubuntu/pool/main/o/openssl/libssl1.1_1.1.1f-1ubuntu2.19_amd64.deb
     dpkg -i libssl1.1_1.1.1f-1ubuntu2.19_amd64.deb
-    git clone https://github.com/intel/neural-speed.git
-    cd neural-speed && pip install -r requirements.txt && pip install .
-    cd ..
-    pip list | grep neural-speed
     python -m pip install --upgrade --force-reinstall torch
     pip install paddlepaddle==2.4.2 paddlenlp==2.5.2 paddlespeech==1.4.1 paddle2onnx==1.0.6
     cd ${WORKING_DIR} || exit 1

diff --git a/intel_extension_for_transformers/neural_chat/requirements.txt b/intel_extension_for_transformers/neural_chat/requirements.txt
@@ -74,3 +74,4 @@ transformers_stream_generator
 qdrant-client
 huggingface_hub
 vllm
+neural_speed
diff --git a/intel_extension_for_transformers/neural_chat/requirements_hpu.txt b/intel_extension_for_transformers/neural_chat/requirements_hpu.txt
@@ -44,3 +44,4 @@ zhconv
 urllib3
 langid
 qdrant-client
+neural_speed
diff --git a/intel_extension_for_transformers/neural_chat/requirements_pc.txt b/intel_extension_for_transformers/neural_chat/requirements_pc.txt
@@ -47,3 +47,4 @@ pymysql
 deepface
 exifread
 qdrant-client
+neural_speed
diff --git a/intel_extension_for_transformers/neural_chat/requirements_xpu.txt b/intel_extension_for_transformers/neural_chat/requirements_xpu.txt
@@ -39,4 +39,5 @@ exifread
 zhconv
 urllib3
 langid
-qdrant-client
+qdrant-client
+neural_speed
diff --git a/intel_extension_for_transformers/neural_chat/tests/requirements.txt b/intel_extension_for_transformers/neural_chat/tests/requirements.txt
@@ -74,3 +74,4 @@ transformers_stream_generator
 qdrant-client
 huggingface_hub
 vllm
+neural_speed
-Original file line number
+Diff line change
@@ Expand Up / @@ -44,3 +44,4 @@ zhconv @@
     urllib3
     langid
     qdrant-client
+    neural_speed