Feature Extraction
Safetensors
English
minicpmv
VisRAG
custom_code
yushi commited on
Commit
3e90d9e
1 Parent(s): e7c5b30

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  license: apache-2.0
3
  datasets:
4
- - test-tcy/VisRAG-Ret-Train-In-domain-data
5
- - test-tcy/VisRAG-Ret-Train-Synthetic-data
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