diff --git a/Dockerfile b/Dockerfile index 983ea00..a12df1e 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 1064b73..299624c 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 c6d3498..ff7ec05 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 00fbd6b..da1efcc 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 6418451..b03851c 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 c30c7df..1146d0a 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 4f36cd7..787120d 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