Update README.md

README.md

@@ -39,6 +39,7 @@ tags:
 
 
 ## 🌎 Model Zoo
+### Vision-Only Checkpoints
 | Model Name     | Type | Visual Encoder | Language Decoder | # Training Frames |
 |:-------------------|:--------------:|:----------------|:------------------|:----------------------:|
 | [VideoLLaMA2-7B-Base](https://huggingface.co/DAMO-NLP-SG/VideoLLaMA2-7B-Base)  | Base  | [clip-vit-large-patch14-336](https://huggingface.co/openai/clip-vit-large-patch14-336) | [Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2)  | 8 |
@@ -52,6 +53,7 @@ tags:
 | [VideoLLaMA2.1-7B-16F-Base](https://huggingface.co/DAMO-NLP-SG/VideoLLaMA2.1-7B-16F-Base)  | Base | [siglip-so400m-patch14-384](https://huggingface.co/google/siglip-so400m-patch14-384) | [Qwen2-7B-Instruct](https://huggingface.co/Qwen/Qwen2-7B-Instruct)  | 16 |
 | [VideoLLaMA2.1-7B-16F](https://huggingface.co/DAMO-NLP-SG/VideoLLaMA2.1-7B-16F)  | Chat | [siglip-so400m-patch14-384](https://huggingface.co/google/siglip-so400m-patch14-384) | [Qwen2-7B-Instruct](https://huggingface.co/Qwen/Qwen2-7B-Instruct)  | 16 |
 
+### Audio-Visual Checkpoints
 | Model Name     | Type | Visual Encoder | Audio Encoder | Language Decoder | # Training Frames |
 |:-------------------|:--------------:|:----------------|:----------------|:------------------|:----------------------:|
 | [VideoLLaMA2.1-7B-AV](https://huggingface.co/DAMO-NLP-SG/VideoLLaMA2.1-7B-AV) (**This Checkpoint**)  | Chat | [siglip-so400m-patch14-384](https://huggingface.co/google/siglip-so400m-patch14-384) |  [Fine-tuned BEATs_iter3+(AS2M)(cpt2)](https://1drv.ms/u/s!AqeByhGUtINrgcpj8ujXH1YUtxooEg?e=E9Ncea) | [VideoLLaMA2.1-7B-16F](https://huggingface.co/DAMO-NLP-SG/VideoLLaMA2.1-7B-16F)  | 8 |
@@ -62,8 +64,6 @@ tags:
 ### Multi-Choice Video QA & Video Captioning
 <p><img src="https://cdn-uploads.huggingface.co/production/uploads/63913b120cf6b11c487ca31d/Z81Dl2MeVlg8wLbYOyTvI.png" width="800" "/></p>
 
-
-
 ###  Open-Ended Video QA
 <p><img src="https://cdn-uploads.huggingface.co/production/uploads/63913b120cf6b11c487ca31d/UoAr7SjbPSPe1z23HBsUh.png" width="800" "/></p>
 
@@ -75,7 +75,7 @@ tags:
 
 
 
-## 🤖 Inference with VideoLLaMA2
+## 🤖 Inference with VideoLLaMA2-AV
 ```python
 import sys
 sys.path.append('./')