diff --git a/vllm/patches/deepseek_v4.py b/vllm/patches/deepseek_v4.py index 46b1d35e..30778e60 100644 --- a/vllm/patches/deepseek_v4.py +++ b/vllm/patches/deepseek_v4.py @@ -1577,14 +1577,13 @@ class DeepseekV4Model(nn.Module): try: weight_loader(param, loaded_weight) except (AssertionError, RuntimeError) as e: - logger.error( - "Failed to load weight %s: param shape=%s, " - "loaded shape=%s, dtype=%s, error=%s", - name, - param.data.shape if hasattr(param, 'data') else '?', - loaded_weight.shape, - loaded_weight.dtype, - e, + print( + f"WEIGHT_LOAD_FAIL: name={name} " + f"param_shape={param.data.shape if hasattr(param, 'data') else '?'} " + f"loaded_shape={loaded_weight.shape} " + f"loaded_dtype={loaded_weight.dtype} " + f"error={e}", + flush=True, ) raise