From 215c58d782a116007c69115718ea0b662bfb6975 Mon Sep 17 00:00:00 2001
From: AlvahLi <15404529+alvahli@user.noreply.gitee.com>
Date: Thu, 30 Jan 2025 14:41:23 +0000
Subject: [PATCH] update download_run_deepseek-coder-6.7b.sh.

Signed-off-by: AlvahLi <15404529+alvahli@user.noreply.gitee.com>
---
 download_run_deepseek-coder-6.7b.sh | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/download_run_deepseek-coder-6.7b.sh b/download_run_deepseek-coder-6.7b.sh
index a221c56..87b8cd1 100755
--- a/download_run_deepseek-coder-6.7b.sh
+++ b/download_run_deepseek-coder-6.7b.sh
@@ -12,14 +12,17 @@ fi
 
 
 # 1，安装 torch 模块，防止依赖多次下载
-# pip3 install torch==2.0.0
+# pip3 install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu121 -i https://pypi.tuna.tsinghua.edu.cn/simple
 
-# 安装 flash_attn 需要网络下载 github 
-# https://www.autodl.com/docs/network_turbo/
+#安装 ninja 和 packaging，这些工具在编译过程中可能会用到
+pip install ninja packaging
 
+# 安装 flash_attn 和 fschat 包
+pip3 install "fschat[model_worker,webui]" flash_attn -i https://mirrors.aliyun.com/pypi/simple
 
-# 最后安装 软件 transformers==4.30.2
-pip3 install "fschat[model_worker,webui]"
+
+# 安装 软件 transformers==4.30.2
+pip install transformers==4.30.2 -i https://mirrors.aliyun.com/pypi/simple
 
 
 # 清除全部 fastchat 服务
-- 
Gitee