From 86255ef376aa71bf05b6e7af0747a8db4fd9f84c Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Tue, 22 Oct 2024 19:55:52 +0800 Subject: [PATCH] fix --- docker/install.sh | 6 ++++-- docker/install_cpu.sh | 4 +++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/docker/install.sh b/docker/install.sh index a76b07de..5c2a49d7 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -19,7 +19,7 @@ curl -fsSL https://ollama.com/install.sh | sh pip install --no-cache-dir -U funasr -pip install --no-cache-dir -U qwen_vl_utils pyav librosa autoawq timm transformers accelerate peft optimum trl safetensors +pip install --no-cache-dir -U qwen_vl_utils pyav librosa timm transformers accelerate peft optimum trl safetensors pip install --no-cache-dir torchsde jupyterlab torchmetrics==0.11.4 tiktoken transformers_stream_generator bitsandbytes basicsr @@ -47,4 +47,6 @@ pip install --no-cache-dir vllm==$vllm_version -U pip install --no-cache-dir -U lmdeploy==$lmdeploy_version --no-deps -pip install --no-cache-dir pynvml shortuuid \ No newline at end of file +pip install --no-cache-dir pynvml shortuuid + +pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version \ No newline at end of file diff --git a/docker/install_cpu.sh b/docker/install_cpu.sh index edac4734..609c94ec 100644 --- a/docker/install_cpu.sh +++ b/docker/install_cpu.sh @@ -20,4 +20,6 @@ pip install --no-cache-dir -U qwen_vl_utils pyav librosa timm transformers accel pip install --no-cache-dir text2sql_lgesql==1.3.0 git+https://github.com/jin-s13/xtcocoapi.git@v1.14 git+https://github.com/gatagat/lap.git@v0.4.0 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html --force --no-deps -pip install --no-cache-dir mpi4py paint_ldm mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 ipykernel fasttext -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html \ No newline at end of file +pip install --no-cache-dir mpi4py paint_ldm mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 ipykernel fasttext -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html + +pip install --no-cache-dir -U torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version --index-url https://download.pytorch.org/whl/cpu \ No newline at end of file