Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,91 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: cc-by-nc-4.0
|
3 |
+
library_name: transformers
|
4 |
+
pipeline_tag: text-generation
|
5 |
+
tags:
|
6 |
+
- NVILA
|
7 |
+
- VLM
|
8 |
+
---
|
9 |
+
|
10 |
+
# VILA Model Card
|
11 |
+
|
12 |
+
## Model details
|
13 |
+
|
14 |
+
**Model type:**
|
15 |
+
NVILA is a visual language model (VLM) pretrained with interleaved image-text data at scale, enabling multi-image VLM. VILA is deployable on the edge, including Jetson Orin and laptop by AWQ 4bit quantization through TinyChat framework. We find: (1) image-text pairs are not enough, interleaved image-text is essential; (2) unfreezing LLM during interleaved image-text pre-training enables in-context learning; (3)re-blending text-only instruction data is crucial to boost both VLM and text-only performance. VILA unveils appealing capabilities, including: multi-image reasoning, in-context learning, visual chain-of-thought, and better world knowledge.
|
16 |
+
|
17 |
+
**Model date:**
|
18 |
+
NVILA was trained in Nov 2024.
|
19 |
+
|
20 |
+
**Paper or resources for more information:**
|
21 |
+
https://github.com/NVLabs/VILA
|
22 |
+
|
23 |
+
```
|
24 |
+
@misc{liu2024nvila,
|
25 |
+
title={NVILA: Efficient Frontier Visual Language Models},
|
26 |
+
author={Zhijian Liu and Ligeng Zhu and Baifeng Shi and Zhuoyang Zhang and Yuming Lou and Shang Yang and Haocheng Xi and Shiyi Cao and Yuxian Gu and Dacheng Li and Xiuyu Li and Yunhao Fang and Yukang Chen and Cheng-Yu Hsieh and De-An Huang and An-Chieh Cheng and Vishwesh Nath and Jinyi Hu and Sifei Liu and Ranjay Krishna and Daguang Xu and Xiaolong Wang and Pavlo Molchanov and Jan Kautz and Hongxu Yin and Song Han and Yao Lu},
|
27 |
+
year={2024},
|
28 |
+
eprint={2412.04468},
|
29 |
+
archivePrefix={arXiv},
|
30 |
+
primaryClass={cs.CV},
|
31 |
+
url={https://arxiv.org/abs/2412.04468},
|
32 |
+
}
|
33 |
+
```
|
34 |
+
|
35 |
+
## License
|
36 |
+
- The code is released under the Apache 2.0 license as found in the [LICENSE](./LICENSE) file.
|
37 |
+
- The pretrained weights are released under the [CC-BY-NC-SA-4.0 license](https://creativecommons.org/licenses/by-nc-sa/4.0/deed.en).
|
38 |
+
- The service is a research preview intended for non-commercial use only, and is subject to the following licenses and terms:
|
39 |
+
- [Terms of Use](https://openai.com/policies/terms-of-use) of the data generated by OpenAI
|
40 |
+
- [Dataset Licenses](https://github.com/Efficient-Large-Model/VILA/blob/main/data_prepare/LICENSE) for each one used during training.
|
41 |
+
|
42 |
+
**Where to send questions or comments about the model:**
|
43 |
+
https://github.com/NVLabs/VILA/issues
|
44 |
+
|
45 |
+
## Intended use
|
46 |
+
**Primary intended uses:**
|
47 |
+
The primary use of VILA is research on large multimodal models and chatbots.
|
48 |
+
|
49 |
+
**Primary intended users:**
|
50 |
+
The primary intended users of the model are researchers and hobbyists in computer vision, natural language processing, machine learning, and artificial intelligence.
|
51 |
+
|
52 |
+
## Input:
|
53 |
+
**Input Type:** Image, Video, Text
|
54 |
+
**Input Format:** Red, Green, Blue; MP4 ;String
|
55 |
+
**Input Parameters:** 2D, 3D
|
56 |
+
|
57 |
+
## Output:
|
58 |
+
**Output Type:** Text
|
59 |
+
**Output Format:** String
|
60 |
+
|
61 |
+
**Supported Hardware Microarchitecture Compatibility:**
|
62 |
+
* Ampere
|
63 |
+
* Jetson
|
64 |
+
* Hopper
|
65 |
+
* Lovelace
|
66 |
+
|
67 |
+
**[Preferred/Supported] Operating System(s):** <br>
|
68 |
+
Linux
|
69 |
+
|
70 |
+
## Training dataset
|
71 |
+
See [Dataset Preparation](https://arxiv.org/abs/2412.04468) for more details.
|
72 |
+
|
73 |
+
** Data Collection Method by dataset
|
74 |
+
* [Hybrid: Automated, Human]
|
75 |
+
|
76 |
+
** Labeling Method by dataset
|
77 |
+
* [Hybrid: Automated, Human]
|
78 |
+
|
79 |
+
## Inference:
|
80 |
+
**Engine:** [Tensor(RT), Triton, Or List Other Here]
|
81 |
+
* PyTorch
|
82 |
+
* TensorRT-LLM
|
83 |
+
* TinyChat
|
84 |
+
|
85 |
+
**Test Hardware:**
|
86 |
+
* A100
|
87 |
+
* Jetson Orin
|
88 |
+
* RTX 4090
|
89 |
+
|
90 |
+
## Ethical Considerations
|
91 |
+
NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
|