Update requirements.txt

czczup · czczup · commit 783535703959 · 2024-07-24T14:33:11.000+08:00
diff --git a/INSTALLATION.md b/INSTALLATION.md
@@ -19,9 +19,31 @@
   pip install -r requirements.txt
   ```
 
-### Additional Instructions for Optional Dependencies
+### Additional Instructions
 
-- Install `apex` (optional, if you want to use `FusedRMSNorm`):
+- Install `flash-attn==2.3.6`:
+
+  ```bash
+  pip install flash-attn==2.3.6 --no-build-isolation
+  ```
+
+  Alternatively you can compile from source:
+
+  ```bash
+  git clone https://github.com/Dao-AILab/flash-attention.git
+  cd flash-attention
+  git checkout v2.3.6
+  python setup.py install
+  ```
+
+- Install `mmcv-full==1.6.2` (optional, for `segmentation`):
+
+  ```bash
+  pip install -U openmim
+  mim install mmcv-full==1.6.2
+  ```
+
+- Install `apex` (optional, for `segmentation`):
 
   ```bash
   git clone https://github.com/NVIDIA/apex.git
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
--r requirements/internvl_chat_training.txt
--r requirements/internvl_chat_eval.txt
--r requirements/internvl_chat_demo.txt
--r requirements/internvl_chat_deploy.txt
+-r requirements/internvl_chat.txt
+-r requirements/streamlit_demo.txt
+-r requirements/classification.txt
+-r requirements/segmentation.txt
diff --git a/requirements/classification.txt b/requirements/classification.txt
@@ -0,0 +1,3 @@
+gdown
+termcolor
+yacs
diff --git a/requirements/clip_benchmark.txt b/requirements/clip_benchmark.txt
@@ -0,0 +1,16 @@
+open_clip_torch>=0.2.1
+opencv-python
+peft>=0.6.2
+protobuf
+pycocoevalcap
+pyyaml
+scikit-learn>=1.0,<2
+scikit-learn
+scipy
+task_adaptation
+tensorflow==2.11.0
+termcolor
+tqdm>=2
+transformers>=4.32.0
+webdataset>=0.2.31
+yacs
diff --git a/requirements/internvl_chat.txt b/requirements/internvl_chat.txt
@@ -5,14 +5,25 @@ deepspeed==0.10.0
 einops==0.6.1
 einops-exts==0.0.4
 flash_attn==2.3.6
+huggingface_hub
 imageio
+lmdeploy
+numpy
 opencv-python
 orjson
 peft>=0.4.0
+pycocoevalcap
+pyyaml
+scikit-learn>=1.2.2
+scipy
 sentencepiece==0.1.99
+shortuuid
 tensorboardX
+termcolor
 timm==0.9.12
 tokenizers==0.15.1
 torch>=2
 torchvision>=0.15
+tqdm
 transformers==4.37.2
+yacs
diff --git a/requirements/internvl_chat_deploy.txt b/requirements/internvl_chat_deploy.txt
diff --git a/requirements/internvl_chat_eval.txt b/requirements/internvl_chat_eval.txt
diff --git a/requirements/segmentation.txt b/requirements/segmentation.txt
@@ -0,0 +1,10 @@
+future
+importlib_metadata
+mmcv-full==1.6.2
+mmsegmentation==0.30.0
+openmim
+ordered-set
+platformdirs
+tensorboard
+tomli
+yapf==0.40.1
diff --git a/requirements/streamlit_demo.txt b/requirements/streamlit_demo.txt
