Views
No views yet
prometheus-eval/prometheus-8x7b-v2.0, produced with llm-compressor for efficient inference in vLLM.| Property | Value |
|---|---|
| Base model | prometheus-eval/prometheus-8x7b-v2.0 (Mixtral 8x7B MoE) |
| Quantization scheme | W4A16 (INT4 weights, FP16 activations) |
| Algorithm | GPTQ |
| Tool | llm-compressor 0.10 |
| Calibration samples | 128 |
| Size on disk | ~26 GB |
| Minimum VRAM for inference | ~28 GB (fits on single L40S / A100 40GB) |
| License | Apache 2.0 (inherited from base model) |
1vllm serve wjacksonrd/prometheus-8x7b-v2.0-W4A16 \
2 --max-model-len 8192 \
3 --gpu-memory-utilization 0.9 \
4 --dtype bfloat161curl http://localhost:8000/v1/completions \
2 -H "Content-Type: application/json" \
3 -d '{
4 "model": "wjacksonrd/prometheus-8x7b-v2.0-W4A16",
5 "prompt": "[INST] ... your judge prompt ... [/INST]",
6 "max_tokens": 1024,
7 "temperature": 0.0
8 }'1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("wjacksonrd/prometheus-8x7b-v2.0-W4A16")
4model = AutoModelForCausalLM.from_pretrained(
5 "wjacksonrd/prometheus-8x7b-v2.0-W4A16",
6 device_map="auto",
7)llm-compressor 0.10.0.1:1from llmcompressor.modifiers.quantization import GPTQModifier
2from llmcompressor import oneshot
3
4recipe = GPTQModifier(
5 targets="Linear",
6 scheme="W4A16",
7 ignore=["lm_head"],
8)
9
10oneshot(
11 model="prometheus-eval/prometheus-8x7b-v2.0",
12 dataset=calibration_dataset,
13 recipe=recipe,
14 num_calibration_samples=128,
15)lm_head is left in full precision.1@article{kim2024prometheus2,
2 title={Prometheus 2: An Open Source Language Model Specialized in Evaluating Other Language Models},
3 author={Kim, Seungone and Suk, Juyoung and Longpre, Shayne and Lin, Bill Yuchen and Shin, Jamin and Welleck, Sean and Neubig, Graham and Lee, Moontae and Lee, Kyungjae and Seo, Minjoon},
4 journal={arXiv preprint arXiv:2405.01535},
5 year={2024}
6}