simon-mo commited on
Commit
3d68a04
·
verified ·
1 Parent(s): 6e71f0f

Remove vLLM FP8 Limitation

Browse files

This has been fixed as of latest v0.8.5 release 🙇

Files changed (1) hide show
  1. README.md +0 -23
README.md CHANGED
@@ -113,29 +113,6 @@ You can use the Qwen3-32B-FP8 model with serveral inference frameworks, includin
113
  However, please pay attention to the following known issues:
114
  - `transformers`:
115
  - there are currently issues with the "fine-grained fp8" method in `transformers` for distributed inference. You may need to set the environment variable `CUDA_LAUNCH_BLOCKING=1` if multiple devices are used in inference.
116
- - vLLM:
117
- - there are currently compatibility issues with `vllm`. For a quick fix, you should make the following changes to `vllm/vllm/model_executor/layers/linear.py`:
118
- ```python
119
- # these changes are in QKVParallelLinear.weight_loader_v2() of vllm/vllm/model_executor/layers/linear.py
120
- ...
121
- shard_offset = self._get_shard_offset_mapping(loaded_shard_id)
122
- shard_size = self._get_shard_size_mapping(loaded_shard_id)
123
-
124
- # add the following code
125
- if isinstance(param, BlockQuantScaleParameter):
126
- weight_block_size = self.quant_method.quant_config.weight_block_size
127
- block_n, _ = weight_block_size[0], weight_block_size[1]
128
- shard_offset = (shard_offset + block_n - 1) // block_n
129
- shard_size = (shard_size + block_n - 1) // block_n
130
- # end of the modification
131
-
132
- param.load_qkv_weight(loaded_weight=loaded_weight,
133
- num_heads=self.num_kv_head_replicas,
134
- shard_id=loaded_shard_id,
135
- shard_offset=shard_offset,
136
- shard_size=shard_size)
137
- ...
138
- ```
139
 
140
  ## Switching Between Thinking and Non-Thinking Mode
141
 
 
113
  However, please pay attention to the following known issues:
114
  - `transformers`:
115
  - there are currently issues with the "fine-grained fp8" method in `transformers` for distributed inference. You may need to set the environment variable `CUDA_LAUNCH_BLOCKING=1` if multiple devices are used in inference.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
116
 
117
  ## Switching Between Thinking and Non-Thinking Mode
118