Fix Qwen neural speed (#1641)

Signed-off-by: changwangss <[email protected]>
intel · Jun 28, 2024 · 76aa9b1 · 76aa9b1
1 parent 63056ec
commit 76aa9b1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/examples/huggingface/neural_speed/requirements.txt b/examples/huggingface/neural_speed/requirements.txt
@@ -1,12 +1,11 @@
 intel_extension_for_transformers
 neural-speed
-lm-eval
+lm-eval==0.4.2
 sentencepiece
 gguf
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch==2.3.0+cpu
 transformers
-intel_extension_for_pytorch==2.3.0
 tiktoken
 transformers_stream_generator
-zipfile38
+zipfile38