software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'main' into privileged

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-02-16 09:29:09 -08:00
parent f886505436 c23aff91d3
commit 88a5d09748
2 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
mods/fix-glm-4.7-flash-AWQ/glm47_vllm_bug.patch Normal file
View File

@@ -0,0 +1,23 @@
--- a/vllm/model_executor/layers/attention/mla_attention.py
+++ b/vllm/model_executor/layers/attention/mla_attention.py
@@ -403,7 +403,7 @@
# If kv_b_proj_weight is unquantized, quantize it to mxfp4 if supported
self.is_aiter_triton_fp4_bmm_enabled = (
rocm_aiter_ops.is_fp4bmm_enabled()
- and self.kv_b_proj.weight.dtype == torch.bfloat16
+ and (self.kv_b_proj.weight.dtype if hasattr(self.kv_b_proj, "weight") else torch.bfloat16) == torch.bfloat16
)
# Attributes for forward_impl method
@@ -2358,9 +2358,9 @@
# model dtype input and will quantize internally.
if (
use_fp8_prefill
- or self.kv_b_proj.weight.dtype != current_platform.fp8_dtype()
+ or (self.kv_b_proj.weight.dtype if hasattr(self.kv_b_proj, "weight") else torch.bfloat16) != current_platform.fp8_dtype()
):
- kv_c_normed = kv_c_normed.to(self.kv_b_proj.weight.dtype)
+ kv_c_normed = kv_c_normed.to((self.kv_b_proj.weight.dtype if hasattr(self.kv_b_proj, "weight") else torch.bfloat16))
k_pe = workspace[:toks][..., self.kv_lora_rank :].unsqueeze(1)
kv_nope = self.kv_b_proj(kv_c_normed)[0].view(

5
mods/fix-glm-4.7-flash-AWQ/run.sh
View File

@@ -1,3 +1,8 @@
#!/bin/bash
set -e
echo "--- Applying GLM 4.7 AWQ speed patch..."
patch -p1 -d / < glm47_flash.patch
echo "=== OK"
echo "--- Applying vLLM crash patch..."
patch -p1 -d /usr/local/lib/python3.12/dist-packages < glm47_vllm_bug.patch || echo "=== Patch is not applicable, skipping"
echo "=== OK"