hekmon commited on
Commit
b0f177c
·
verified ·
1 Parent(s): 1566d8d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -1
README.md CHANGED
@@ -4,4 +4,27 @@ library_name: transformers
4
  pipeline_tag: image-text-to-text
5
  base_model:
6
  - numind/NuExtract-2.0-8B
7
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  pipeline_tag: image-text-to-text
5
  base_model:
6
  - numind/NuExtract-2.0-8B
7
+ ---
8
+
9
+ # NuExtract-2.0-8B-FP8-Dynamic
10
+
11
+ ## Quantization
12
+
13
+ Quantized with [llm-compressor](https://github.com/vllm-project/llm-compressor) v0.9.0.1.
14
+
15
+ We used the original [qwen2.5-vl example compression script](https://github.com/vllm-project/llm-compressor/blob/main/examples/multimodal_vision/qwen_2_5_vl_example.py) and adapted it to a [FP8-Dynamic compression recipe](https://github.com/vllm-project/llm-compressor/tree/main/examples/quantization_w8a8_fp8).
16
+
17
+ ## vLLM inference
18
+
19
+ ```bash
20
+ docker run --rm --name 'NuExtract-2.0-8B' -e HF_TOKEN -v '/srv/cache:/root/.cache' -p 8000:8000 -e LD_LIBRARY_PATH='/lib/x86_64-linux-gnu:/usr/local/cuda/lib64' 'vllm/vllm-openai:v0.15.1-cu130' 'ig1/NuExtract-2.0-8B-FP8-Dynamic' --served-model-name 'NuExtract-2.0-8B' --trust-remote-code --limit-mm-per-prompt '{"image": 6, "video": 0}' --chat-template-content-format 'openai' --max-model-len 'auto' --kv-cache-memory-bytes '7G'
21
+ ```
22
+
23
+ * `-e LD_LIBRARY_PATH=/lib64:/usr/local/cuda/lib64` is only needed if your host have a recent driver version (with native CUDA 13.0 or 13.1). See [#32373](https://github.com/vllm-project/vllm/issues/32373) for more info.
24
+ * Adapt `/srv/cache` to your liking, this will contains all cache data you want to keep for faster startup:
25
+ * dirs like `huggingface`, `torch`, `vllm`, `flashinfer`, etc...
26
+ * To avoid eating up all the GPU VRAM, the `--kv-cache-memory-bytes '7G'` is set (it allows max model len by 1.02x)
27
+ * Feel free to adjust (or remove the flag and switch back to `--gpu-memory-utilization 0.9`) to increase or decrease KV cache to your liking
28
+
29
+
30
+ Check original project readme for openai like chat template use in requests.