From 06e6311a0c4fd3b11e4fec78a5585a95182d9f35 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Thu, 30 Apr 2026 23:47:18 +0800 Subject: [PATCH] fix --- docker/install.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/install.sh b/docker/install.sh index b39f2137..0d593202 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -10,7 +10,7 @@ flashattn_version=${7:-2.7.1.post4} pip uninstall -y torch torchvision torchaudio -pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version --index-url https://download.pytorch.org/whl/cu129 +pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version --index-url https://download.pytorch.org/whl/cu130 pip install --no-cache-dir tiktoken transformers_stream_generator bitsandbytes deepspeed torchmetrics decord optimum openai-whisper