forked from huggingface/tgi-gaudi
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support for GPTQ Marlin (huggingface#2052)
Add support for GPTQ Marlin kernels GPTQ Marlin extends the Marlin kernels to support common GPTQ configurations: - bits: 4 or 8 - groupsize: -1, 32, 64, or 128 - desc_act: true/false Using the GPTQ Marlin kernels requires repacking the parameters in the Marlin quantizer format. The kernels were contributed by Neural Magic to VLLM. We vendor them here for convenience.
- Loading branch information
1 parent
7ce29b1
commit f1f2840
Showing
39 changed files
with
4,649 additions
and
135 deletions.
There are no files selected for viewing
84 changes: 84 additions & 0 deletions
84
...tests/models/__snapshots__/test_flash_llama_gptq_marlin/test_flash_llama_gptq_marlin.json
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,84 @@ | ||
{ | ||
"details": { | ||
"best_of_sequences": null, | ||
"finish_reason": "length", | ||
"generated_tokens": 10, | ||
"prefill": [ | ||
{ | ||
"id": 2323, | ||
"logprob": null, | ||
"text": "Test" | ||
}, | ||
{ | ||
"id": 1715, | ||
"logprob": -11.34375, | ||
"text": " request" | ||
} | ||
], | ||
"seed": null, | ||
"tokens": [ | ||
{ | ||
"id": 198, | ||
"logprob": -2.5742188, | ||
"special": false, | ||
"text": "\n" | ||
}, | ||
{ | ||
"id": 262, | ||
"logprob": -1.6230469, | ||
"special": false, | ||
"text": " " | ||
}, | ||
{ | ||
"id": 3270, | ||
"logprob": -2.046875, | ||
"special": false, | ||
"text": " \"\"\"\n" | ||
}, | ||
{ | ||
"id": 262, | ||
"logprob": -0.015281677, | ||
"special": false, | ||
"text": " " | ||
}, | ||
{ | ||
"id": 422, | ||
"logprob": -2.1425781, | ||
"special": false, | ||
"text": " if" | ||
}, | ||
{ | ||
"id": 1715, | ||
"logprob": -0.9238281, | ||
"special": false, | ||
"text": " request" | ||
}, | ||
{ | ||
"id": 13204, | ||
"logprob": -0.076660156, | ||
"special": false, | ||
"text": ".method" | ||
}, | ||
{ | ||
"id": 624, | ||
"logprob": -0.021987915, | ||
"special": false, | ||
"text": " ==" | ||
}, | ||
{ | ||
"id": 364, | ||
"logprob": -0.39208984, | ||
"special": false, | ||
"text": " '" | ||
}, | ||
{ | ||
"id": 3019, | ||
"logprob": -0.10821533, | ||
"special": false, | ||
"text": "POST" | ||
} | ||
], | ||
"top_tokens": null | ||
}, | ||
"generated_text": "\n \"\"\"\n if request.method == 'POST" | ||
} |
84 changes: 84 additions & 0 deletions
84
...s/__snapshots__/test_flash_llama_gptq_marlin/test_flash_llama_gptq_marlin_all_params.json
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,84 @@ | ||
{ | ||
"details": { | ||
"best_of_sequences": null, | ||
"finish_reason": "length", | ||
"generated_tokens": 10, | ||
"prefill": [ | ||
{ | ||
"id": 2323, | ||
"logprob": null, | ||
"text": "Test" | ||
}, | ||
{ | ||
"id": 1715, | ||
"logprob": -11.34375, | ||
"text": " request" | ||
} | ||
], | ||
"seed": 0, | ||
"tokens": [ | ||
{ | ||
"id": 13, | ||
"logprob": -2.2539062, | ||
"special": false, | ||
"text": "." | ||
}, | ||
{ | ||
"id": 578, | ||
"logprob": -0.15563965, | ||
"special": false, | ||
"text": " The" | ||
}, | ||
{ | ||
"id": 3622, | ||
"logprob": -0.8203125, | ||
"special": false, | ||
"text": " server" | ||
}, | ||
{ | ||
"id": 706, | ||
"logprob": 0.0, | ||
"special": false, | ||
"text": " has" | ||
}, | ||
{ | ||
"id": 539, | ||
"logprob": 0.0, | ||
"special": false, | ||
"text": " not" | ||
}, | ||
{ | ||
"id": 3686, | ||
"logprob": 0.0, | ||
"special": false, | ||
"text": " yet" | ||
}, | ||
{ | ||
"id": 3288, | ||
"logprob": 0.0, | ||
"special": false, | ||
"text": " sent" | ||
}, | ||
{ | ||
"id": 904, | ||
"logprob": 0.0, | ||
"special": false, | ||
"text": " any" | ||
}, | ||
{ | ||
"id": 828, | ||
"logprob": 0.0, | ||
"special": false, | ||
"text": " data" | ||
}, | ||
{ | ||
"id": 382, | ||
"logprob": -1.5517578, | ||
"special": false, | ||
"text": ".\n\n" | ||
} | ||
], | ||
"top_tokens": null | ||
}, | ||
"generated_text": "Test request. The server has not yet sent any data.\n\n" | ||
} |
Oops, something went wrong.