marcellacornia's picture
Update README.md
b98f693 verified
---
license: apache-2.0
datasets:
- liuhaotian/LLaVA-CC3M-Pretrain-595K
library_name: transformers
pipeline_tag: image-text-to-text
---
# Model Card: LLaVA_MORE-llama_3_1-8B-pretrain
```LLaVA-MORE``` enhances the well-known LLaVA architecture by integrating the use of LLaMA 3.1 as the language model. We are publicly releasing the checkpoints for stages one and two for the first model with 8B parameters.
In this model space, you will find the stage one (pretrain) weights of LLaVA-MORE LLaMA 3.1 8B.
For more information, visit our [LLaVA-MORE](https://github.com/aimagelab/LLaVA-MORE) repository.
## Inference
You can try our LLaVA-MORE in the Image-To-Text task by cloning our repository and running the following script.
```bash
python -u llava/eval/run_llava.py
```
## Citation
If you make use of our work, please cite our repo:
```bibtex
@misc{cocchi2024llavamore,
title={{LLaVA-MORE: Enhancing Visual Instruction Tuning with LLaMA 3.1}},
author={Cocchi, Federico and Moratelli, Nicholas and Caffagni, Davide and Sarto, Sara and Cornia, Marcella and Baraldi, Lorenzo and Cucchiara, Rita},
url={https://github.com/aimagelab/LLaVA-MORE},
year={2024}
}
```