From 3dd545a79fbe6ec2891f6d7cf7fcd181e06f6189 Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Thu, 23 May 2024 07:41:51 +0000 Subject: [PATCH] format --- docs/installation.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/installation.md b/docs/installation.md index c96161c..71a36cd 100644 --- a/docs/installation.md +++ b/docs/installation.md @@ -1,8 +1,9 @@ # Installation Requirements are listed in `requirements` folder. +Note that besides these packages, some packages needs to be mannually installed, and are detailed in the following sections. -## Different CUDA versions +# Different CUDA versions You need to manually install `torch`, `torchvision` and `xformers` for different CUDA versions. @@ -35,7 +36,7 @@ pip install flash-attn --no-build-isolation pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex.git ``` -## Different Dependencies +# Optional Dependencies The default installation is for inference only. Other optional dependencies are detailed in sections below, namel "Data Dependencies" and "Evaluation Dependencies".