Text Generation
Transformers
Safetensors
llava_llama
VILA
VLM
nielsr HF Staff commited on
Commit
d0b1731
·
verified ·
1 Parent(s): d6bce15

Refine Model Card: QServe Information and Links

Browse files

This PR updates the model card to accurately reflect the QServe model and its capabilities, replacing the outdated VILA information. The `pipeline_tag` is updated to `text-generation`, and key features from the paper and README are summarized.

Files changed (1) hide show
  1. README.md +23 -105
README.md CHANGED
@@ -1,37 +1,37 @@
1
  ---
2
- license: cc-by-nc-4.0
3
  library_name: transformers
 
4
  pipeline_tag: text-generation
5
  tags:
6
- - VILA
7
- - VLM
 
8
  ---
9
 
10
- # VILA Model Card
11
 
12
- ## Model details
13
 
14
- **Model type:**
15
- VILA 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
- VILA1.5-13b was trained in May 2024.
19
 
20
- **Paper or resources for more information:**
21
- https://github.com/NVLabs/VILA
22
 
23
- ```
24
- @misc{lin2023vila,
25
- title={VILA: On Pre-training for Visual Language Models},
26
- author={Ji Lin and Hongxu Yin and Wei Ping and Yao Lu and Pavlo Molchanov and Andrew Tao and Huizi Mao and Jan Kautz and Mohammad Shoeybi and Song Han},
27
- year={2023},
28
- eprint={2312.07533},
29
- archivePrefix={arXiv},
30
- primaryClass={cs.CV}
31
- }
32
- ```
 
 
 
33
 
34
- https://github.com/mit-han-lab/qserve
35
  ```
36
  @article{lin2024qserve,
37
  title={QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving},
@@ -39,86 +39,4 @@ https://github.com/mit-han-lab/qserve
39
  journal={arXiv preprint arXiv:2405.04532},
40
  year={2024}
41
  }
42
- ```
43
-
44
- ## License
45
- - The code is released under the Apache 2.0 license as found in the [LICENSE](./LICENSE) file.
46
- - 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).
47
- - The service is a research preview intended for non-commercial use only, and is subject to the following licenses and terms:
48
- - [Model License](https://github.com/facebookresearch/llama/blob/main/MODEL_CARD.md) of LLaMA
49
- - [Terms of Use](https://openai.com/policies/terms-of-use) of the data generated by OpenAI
50
- - [Dataset Licenses](https://github.com/Efficient-Large-Model/VILA/blob/main/data_prepare/LICENSE) for each one used during training.
51
-
52
- **Where to send questions or comments about the model:**
53
- https://github.com/NVLabs/VILA/issues
54
-
55
- ## Intended use
56
- **Primary intended uses:**
57
- The primary use of VILA is research on large multimodal models and chatbots.
58
-
59
- **Primary intended users:**
60
- The primary intended users of the model are researchers and hobbyists in computer vision, natural language processing, machine learning, and artificial intelligence.
61
-
62
- ## Model Architecture:
63
- **Architecture Type:** Transformer
64
- **Network Architecture:** siglip, vicuna1.5
65
-
66
- ## Input:
67
- **Input Type:** Image, Video, Text
68
- **Input Format:** Red, Green, Blue; MP4 ;String
69
- **Input Parameters:** 2D, 3D
70
-
71
- ## Output:
72
- **Output Type:** Text
73
- **Output Format:** String
74
-
75
- **Supported Hardware Microarchitecture Compatibility:**
76
- * Ampere
77
- * Jetson
78
- * Hopper
79
- * Lovelace
80
-
81
- **[Preferred/Supported] Operating System(s):** <br>
82
- Linux
83
-
84
- ## Model Version(s):
85
- * VILA1.5-3B
86
- * VILA1.5-3B-s2
87
- * Llama-3-VILA1.5-8B
88
- * VILA1.5-13B
89
- * VILA1.5-40B
90
- * VILA1.5-3B-AWQ
91
- * VILA1.5-3B-s2-AWQ
92
- * Llama-3-VILA1.5-8B-AWQ
93
- * VILA1.5-13B-AWQ
94
- * VILA1.5-40B-AWQ
95
-
96
- ## Training dataset
97
- See [Dataset Preparation](https://github.com/NVLabs/VILA/blob/main/data_prepare/README.md) for more details.
98
-
99
- ** Data Collection Method by dataset
100
- * [Hybrid: Automated, Human]
101
-
102
- ** Labeling Method by dataset
103
- * [Hybrid: Automated, Human]
104
-
105
- **Properties (Quantity, Dataset Descriptions, Sensor(s)):**
106
- 53 million image-text pairs or interleaved image text content.
107
-
108
-
109
- ## Evaluation dataset
110
- A collection of 12 benchmarks, including 5 academic VQA benchmarks and 7 recent benchmarks specifically proposed for instruction-following LMMs.
111
-
112
- ## Inference:
113
- **Engine:** [Tensor(RT), Triton, Or List Other Here]
114
- * PyTorch
115
- * TensorRT-LLM
116
- * TinyChat
117
-
118
- **Test Hardware:**
119
- * A100
120
- * Jetson Orin
121
- * RTX 4090
122
-
123
- ## Ethical Considerations
124
- 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.
 
1
  ---
 
2
  library_name: transformers
3
+ license: cc-by-nc-4.0
4
  pipeline_tag: text-generation
5
  tags:
6
+ - quantization
7
+ - llm-serving
8
+ - efficient-inference
9
  ---
10
 
11
+ # QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving
12
 
13
+ This repository contains the QServe inference library and quantized model checkpoints as described in [QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving](https://huggingface.co/papers/2405.04532). QServe significantly improves the throughput of large language model serving on GPUs through W4A8KV4 quantization and system co-design.
14
 
15
+ Code: https://github.com/mit-han-lab/qserve
 
16
 
17
+ ## Key Features
 
18
 
19
+ QServe offers:
 
20
 
21
+ * **High Throughput:** Achieves significant speedups compared to TensorRT-LLM, especially on cost-effective GPUs like the L40S. Improves maximum achievable throughput of Llama-3-8B by 1.2x on A100, 1.4x on L40S; and Qwen1.5-72B by 2.4x on A100, 3.5x on L40S.
22
+ * **W4A8KV4 Quantization:** Employs 4-bit weights, 8-bit activations, and 4-bit KV cache for efficient inference.
23
+ * **QoQ Quantization Algorithm:** Implements the quattuor-octo-quattuor (4-8-4) quantization algorithm.
24
+ * **Low Dequantization Overhead:** Minimizes runtime overhead associated with dequantization using progressive quantization.
25
+ * **SmoothAttention:** Mitigates accuracy degradation from 4-bit KV quantization.
26
+ * **Compute-Aware Weight Reordering and Register-Level Parallelism:** Reduces dequantization latency.
27
+ * **Memory-Bound Fused Attention:** Leverages the performance gains of KV4 quantization.
28
+ * **PyTorch-Based:** Provides PyTorch-level flexibility while maintaining TensorRT-LLM-level efficiency.
29
+ * **Full Support for In-flight Batching and Paged Attention:** Enables efficient handling of large requests and long sequences.
30
+ * **Large-Scale Generation Support:** Enables efficient large-scale synthetic data generation.
31
+
32
+
33
+ ## Citation
34
 
 
35
  ```
36
  @article{lin2024qserve,
37
  title={QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving},
 
39
  journal={arXiv preprint arXiv:2405.04532},
40
  year={2024}
41
  }
42
+ ```