Merge pull request #19 from smalltong02/RAG_dev1.0.1

update readme.md
smalltong02 · Feb 2, 2024 · 3869945 · 3869945
2 parents 7fcdeb6 + 5b30a73
commit 3869945
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/readme-cn.md b/readme-cn.md
@@ -149,7 +149,7 @@
   1. 如果拥有NVIDIA GPU，请首先安装CUDA Toolkit (https://developer.nvidia.com/cuda-toolkit-archive) ，并在虚拟环境中安装pytorch CUDA版本 (版本号请和CUDA Toolkit版本相同 https://pytorch.org/)
   ```bash
   // 例如安装12.1版本
-  conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
+  conda install pytorch=2.1.2 torchvision=0.16.2 torchaudio=2.1.2 pytorch-cuda=12.1 -c pytorch -c nvidia
   ```
 
   1. 安装依赖项, 请按照不同平台选择适当的requirements

diff --git a/readme.md b/readme.md
@@ -151,7 +151,7 @@ Auxiliary models, such as retrieval, code execution, text-to-speech, speech-to-t
   1. If you have an NVIDIA GPU, Please install the CUDA Toolkit from (https://developer.nvidia.com/cuda-toolkit-archive), and install the PyTorch CUDA version in the virtual environment (same to the CUDA Toolkit version https://pytorch.org/):
   ```bash
   // such as install version 12.1
-  conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
+  conda install pytorch=2.1.2 torchvision=0.16.2 torchaudio=2.1.2 pytorch-cuda=12.1 -c pytorch -c nvidia
   ```
 
   1. Install dependencies, Please choose the appropriate requirements file based on your platform, On the Windows, if encounter compilation errors for llama-cpp-python or tts during the installation, please remove these two packages from the requirements:

diff --git a/requirements-windows.txt b/requirements-windows.txt
@@ -136,4 +136,4 @@ zipp==3.17.0
 SpeechRecognition==3.10.0
 
 xformers==0.0.23.post1
-diffusers==0.25.1
+diffusers==0.25.1