Fix transformers version dependabot issues (#5174)

* fix dependabot issues * fix
intel · Apr 11, 2024 · b4cde76 · b4cde76
1 parent 6b413f1
commit b4cde76
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 6 deletions.
diff --git a/ppml/tdx/docker/trusted-bigdl-llm/inference/Dockerfile b/ppml/tdx/docker/trusted-bigdl-llm/inference/Dockerfile
@@ -48,6 +48,7 @@ RUN env DEBIAN_FRONTEND=noninteractive apt-get update && \
     pip3 install -r /opt/requirements.txt && \
     pip3 install typing-inspect==0.8.0 typing_extensions==4.5.0 && \
     pip3 install ray==2.3.1 && \
+    pip3 install transformers==4.36.0 && \
     pip3 install sentence-transformers --no-deps && \
     pip3 install cmake ninja cpuid && \
     pip3 install scikit-build && \

diff --git a/ppml/tdx/docker/trusted-bigdl-llm/inference/requirements.txt b/ppml/tdx/docker/trusted-bigdl-llm/inference/requirements.txt
@@ -1,4 +1,3 @@
-transformers==4.36.0
 # install torch first:
 # pip install torch==2.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu
 

diff --git a/python/llm/example/CPU/Speculative-Decoding/baichuan2/README.md b/python/llm/example/CPU/Speculative-Decoding/baichuan2/README.md
@@ -73,7 +73,7 @@ python -m pip install oneccl_bind_pt==2.2.0 --extra-index-url https://pytorch-ex
 # if there is any installation problem for oneccl_binding, you can also find suitable index url at "https://pytorch-extension.intel.com/release-whl/stable/cpu/cn/" or "https://developer.intel.com/ipex-whl-stable-cpu" according to your environment.
 
 # Install other dependencies
-pip install -r requirements.txt
+python -m pip install transformers==4.36.2 transformers-stream-generator
 ```
 
 #### 4.2 Run Baichuan2 Models with IPEX

diff --git a/python/llm/example/CPU/Speculative-Decoding/baichuan2/requirements.txt b/python/llm/example/CPU/Speculative-Decoding/baichuan2/requirements.txt
diff --git a/python/llm/example/CPU/Speculative-Decoding/vicuna/requirements.txt b/python/llm/example/CPU/Speculative-Decoding/vicuna/requirements.txt