Update README.md

README.md

@@ -31,13 +31,29 @@ tags:
 datasets:
 - wikitext
 ---
-
-# Important Note
-- Two files are modified to address a current issue regarding Llama-3s keeps on generating additional tokens non-stop until hitting max token limit.
-- `generation_config.json`'s `eos_token_id` have been modified to add the other EOS token that Llama-3 uses
-- `tokenizer_config.json`'s `chat_template` has been modified to only add start generation token at the end of a prompt if `add_generation_prompt` is selected
-
-
+<!-- header start -->
+<!-- 200823 -->
+<div style="width: auto; margin-left: auto; margin-right: auto">
+<img src="https://www.astronomer.io/logo/astronomer-logo-RGB-standard-1200px.png" alt="Astronomer" style="width: 60%; min-width: 400px; display: block; margin: auto;">
+</div>
+<div style="display: flex; justify-content: space-between; width: 100%;">
+    <div style="display: flex; flex-direction: column; align-items: flex-start;">
+        <p style="margin-top: 0.5em; margin-bottom: 0em;"></p>
+    </div>
+    <div style="display: flex; flex-direction: column; align-items: flex-end;">
+        <p style="margin-top: 0.5em; margin-bottom: 0em;"><a href="https://www.linkedin.com/in/david-xue-uva/">Quantized by David Xue, ML Engineer @ Astronomer</a></p>
+    </div>
+</div>
+<div style="text-align:center; margin-top: 0em; margin-bottom: 0em"><p style="margin-top: 0.25em; margin-bottom: 0em;">This model is generously created and made open source by <a href="https://astronomer.io">Astronomer</a></p></div>
+<hr style="margin-top: 1.0em; margin-bottom: 1.0em;">
+<!-- header end -->
+# Important Note Regarding a Known Bug in Llama 3
+- Two files are modified to address a current issue regarding Llama 3 models keep on generating additional tokens non-stop until hitting max token limit.
+- `generation_config.json`'s `eos_token_id` have been modified to add the other EOS token that Llama-3 uses.
+- `tokenizer_config.json`'s `chat_template` has been modified to only add start generation token at the end of a prompt if `add_generation_prompt` is selected.
+- For loading this model onto vLLM, make sure all requests have `"stop_token_ids":[128001, 128009]` to temporarily address the non-stop generation issue.
+   - vLLM does not yet respect `generation_config.json`.
+   - vLLM team is working on a a fix for this https://github.com/vllm-project/vllm/issues/4180
 
 # Llama-3-8B-Instruct-GPTQ-8-Bit
 - Model creator: [Meta Llama from Meta](https://huggingface.co/meta-llama)
@@ -52,7 +68,26 @@ This repo contains 8 Bit quantized GPTQ model files for [meta-llama/Meta-Llama-3
 <!-- description end -->
 
 ## GPTQ Quantization Method
-| Branch | Bits | GS | Act Order | Damp % | GPTQ Dataset | Seq Len | Size | ExLlama | Desc |
+| Branch | Bits | GS | Act Order | Damp % | GPTQ Dataset | Seq Len | VRAM Size | ExLlama | Desc |
 | ------ | ---- | -- | --------- | ------ | ------------ | ------- | ---- | ------- | ---- |
-| [main](https://huggingface.co/TheBloke/Mistral-7B-v0.1-GPTQ/tree/main) | 8 | 32 | Yes | 0.1 | [wikitext](https://huggingface.co/datasets/wikitext/viewer/wikitext-2-v1/test) | 32768 | 4.16 GB | No | 8-bit, with Act Order and group size 32g. Minimum accuracy loss with decent VRAM usage reduction. | 
-| More variants to come | TBD | TBD | TBD | TBD | TBD | TBD | TBD | TBD | May upload additional variants of GPTQ 8 bit models in the future using different parameters such as 128g group size and etc. |
+| [main](https://huggingface.co/astronomer-io/Llama-3-8B-Instruct-GPTQ-8-Bit/tree/main) | 8 | 32 | Yes | 0.1 | [wikitext](https://huggingface.co/datasets/wikitext/viewer/wikitext-2-v1/test) | 8192 | 9.09 GB | No | 8-bit, with Act Order and group size 32g. Minimum accuracy loss with decent VRAM usage reduction. | 
+| More variants to come | TBD | TBD | TBD | TBD | TBD | TBD | TBD | TBD | May upload additional variants of GPTQ 8 bit models in the future using different parameters such as 128g group size and etc. |
+
+## Serving this GPTQ model using vLLM
+Tested with the below command
+```
+python -m vllm.entrypoints.openai.api_server --model Llama-3-8B-Instruct-GPTQ-8-Bit --port 8123 --max-model-len 8192 --dtype float16
+```
+For the non-stop token generation bug, make sure to send requests with `stop_token_ids":[128001, 128009]` to vLLM endpoint
+Example:
+```
+{
+    "model": "Llama-3-8B-Instruct-GPTQ-8-Bit",
+    "messages": [
+        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "user", "content": "Who created Llama 3?"}
+        ],
+    "max_tokens": 2000,
+    "stop_token_ids":[128001,128009]
+}
+```