diff --git a/download_run_deepseek-coder-6.7b.sh b/download_run_deepseek-coder-6.7b.sh index a221c568f171e2fb881788a686b5ac694d76e25b..87b8cd1e7dcacea543373ddc50d1fb19f98250d1 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 服务