Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix install.sh using bash in Dockerfile.ubuntu

Browse Source
This commit is contained in:
wangxingjun778
2025-09-16 22:46:58 +08:00
parent d5a60d0ba2
commit 73ae750b73
2 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile.ubuntu
View File

@@ -56,7 +56,7 @@ fi
ARG CUR_TIME={cur_time}
RUN echo $CUR_TIME
RUN sh /tmp/install.sh {version_args} && \
RUN bash /tmp/install.sh {version_args} && \
curl -fsSL https://ollama.com/install.sh | sh && \
pip install --no-cache-dir -U funasr scikit-learn && \
pip install --no-cache-dir -U qwen_vl_utils qwen_omni_utils librosa timm transformers accelerate peft trl safetensors && \

15
docker/install.sh
View File

@@ -25,14 +25,11 @@ pip install --no-cache-dir flash_attn==$flashattn_version
pip install --no-cache-dir triton auto-gptq==$autogptq_version -U && pip cache purge
pip install --no-cache-dir vllm==$vllm_version && pip cache purge
#if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0" ]]; then
# # vllm_version is >= 0.6.0
# echo ">>Debug: Condition met: vllm_version >= 0.6.0, proceeding with installation..."
# pip install --no-cache-dir vllm==$vllm_version && pip cache purge
#else
# echo ">>Debug: Condition not met: vllm_version < 0.6.0, skipping installation. (vllm_version = $vllm_version)"
#fi
if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0" ]]; then
# vllm_version is >= 0.6.0
pip install --no-cache-dir vllm==$vllm_version && pip cache purge
else
echo "vllm_version < 0.6.0, skipping installation. (vllm_version = $vllm_version)"
fi
# pip uninstall -y torch-scatter && TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9;9.0" pip install --no-cache-dir -U torch-scatter