Update README.md
Browse files
README.md
CHANGED
@@ -1,8 +1,8 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
datasets:
|
4 |
-
-
|
5 |
-
-
|
6 |
language:
|
7 |
- en
|
8 |
base_model:
|
@@ -31,9 +31,6 @@ Our training dataset of 362,110 Query-Document (Q-D) Pairs for **VisRAG-Ret** is
|
|
31 |
### VisRAG-Gen
|
32 |
The generation part does not use any fine-tuning; we directly use off-the-shelf LLMs/VLMs for generation.
|
33 |
|
34 |
-
## Implementation Details
|
35 |
-
**VisRAG-Ret** is fine-tuned using [in-batch negatives](https://arxiv.org/abs/2004.04906) for one epoch with a batch size of 128 on 8 NVIDIA A100 80GB GPUs. The temperature is set to 0.02.
|
36 |
-
|
37 |
## Requirements
|
38 |
```
|
39 |
torch==2.1.2
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
datasets:
|
4 |
+
- openbmb/VisRAG-Ret-Train-In-domain-data
|
5 |
+
- openbmb/VisRAG-Ret-Train-Synthetic-data
|
6 |
language:
|
7 |
- en
|
8 |
base_model:
|
|
|
31 |
### VisRAG-Gen
|
32 |
The generation part does not use any fine-tuning; we directly use off-the-shelf LLMs/VLMs for generation.
|
33 |
|
|
|
|
|
|
|
34 |
## Requirements
|
35 |
```
|
36 |
torch==2.1.2
|