diff --git a/Dockerfile b/Dockerfile index 983ea003ea..a12df1e0b7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -54,7 +54,7 @@ RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi python3 -m pip cache purge # Install modelscope (for fast download) and ray (for multinode) -RUN python3 -m pip install modelscope ray && \ +RUN python3 -m pip install modelscope 'ray>=2.47.1' 'protobuf>3.20.0' && \ python3 -m pip cache purge CMD ["/bin/bash"] diff --git a/Dockerfile.310p b/Dockerfile.310p index 1064b73ed2..299624c541 100644 --- a/Dockerfile.310p +++ b/Dockerfile.310p @@ -55,7 +55,7 @@ RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi python3 -m pip cache purge # Install modelscope (for fast download) and ray (for multinode) -RUN python3 -m pip install modelscope ray && \ +RUN python3 -m pip install modelscope 'ray>=2.47.1' 'protobuf>3.20.0' && \ python3 -m pip cache purge CMD ["/bin/bash"] diff --git a/Dockerfile.310p.openEuler b/Dockerfile.310p.openEuler index c6d34984f1..ff7ec05a43 100644 --- a/Dockerfile.310p.openEuler +++ b/Dockerfile.310p.openEuler @@ -52,7 +52,7 @@ RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi python3 -m pip cache purge # Install modelscope (for fast download) and ray (for multinode) -RUN python3 -m pip install modelscope ray && \ +RUN python3 -m pip install modelscope 'ray>=2.47.1' 'protobuf>3.20.0' && \ python3 -m pip cache purge CMD ["/bin/bash"] diff --git a/Dockerfile.a3 b/Dockerfile.a3 index 00fbd6b4df..da1efcc41b 100644 --- a/Dockerfile.a3 +++ b/Dockerfile.a3 @@ -54,7 +54,7 @@ RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi python3 -m pip cache purge # Install modelscope (for fast download) and ray (for multinode) -RUN python3 -m pip install modelscope ray && \ +RUN python3 -m pip install modelscope 'ray>=2.47.1' 'protobuf>3.20.0' && \ python3 -m pip cache purge CMD ["/bin/bash"] \ No newline at end of file diff --git a/Dockerfile.a3.openEuler b/Dockerfile.a3.openEuler index 64184518ed..b03851ca65 100644 --- a/Dockerfile.a3.openEuler +++ b/Dockerfile.a3.openEuler @@ -51,7 +51,7 @@ RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi python3 -m pip cache purge # Install modelscope (for fast download) and ray (for multinode) -RUN python3 -m pip install modelscope ray && \ +RUN python3 -m pip install modelscope 'ray>=2.47.1' 'protobuf>3.20.0' && \ python3 -m pip cache purge CMD ["/bin/bash"] \ No newline at end of file diff --git a/Dockerfile.openEuler b/Dockerfile.openEuler index c30c7df68b..1146d0a00a 100644 --- a/Dockerfile.openEuler +++ b/Dockerfile.openEuler @@ -51,7 +51,7 @@ RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi python3 -m pip cache purge # Install modelscope (for fast download) and ray (for multinode) -RUN python3 -m pip install modelscope ray && \ +RUN python3 -m pip install modelscope 'ray>=2.47.1' 'protobuf>3.20.0' && \ python3 -m pip cache purge CMD ["/bin/bash"] diff --git a/requirements-dev.txt b/requirements-dev.txt index 4f36cd70d9..787120d757 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -14,6 +14,6 @@ pytest-cov regex sentence_transformers ray>=2.47.1 -protobuf==4.25.6 +protobuf>3.20.0 librosa soundfile