more fixes

2026-05-14 19:39:16 +00:00
parent 1ceff541b0
commit 4749a92fca
1 changed files with 1 additions and 1 deletions
--- a/vllm/patches/deepseek_v4.py
+++ b/vllm/patches/deepseek_v4.py
@@ -208,7 +208,7 @@ class DeepseekV4FP8Config(Fp8Config):

 # Staging kernel imported from standalone module (has proper RNE rounding
 # and subnormal handling for UE4M3 — the old embedded copy was broken).
-from staging_kernel import _stage_deepseek_v4_mega_moe_inputs
+from vllm.model_executor.models.staging_kernel import _stage_deepseek_v4_mega_moe_inputs


 def make_deepseek_v4_expert_params_mapping(