[ci] fix broken tests (vllm-project#10956)

Signed-off-by: youkaichao <[email protected]>

vllm/worker/model_runner.py

@@ -1782,16 +1782,16 @@ def need_recv_kv(self, model_input, kv_caches) -> bool:
             kv_caches: vLLM's paged memory
         """
 
+        if self.vllm_config.kv_transfer_config is None:
+            return False
+
         prefill_meta = model_input.attn_metadata.prefill_metadata
 
         # check if the current run is profiling
         is_profile_run = (kv_caches[0].numel() == 0)
         # check if the current run is prefill
         is_prefill_run = prefill_meta is not None
 
-        if self.vllm_config.kv_transfer_config is None:
-            return False
-
         return self.vllm_config.kv_transfer_config.is_kv_consumer and (
             not is_profile_run) and is_prefill_run
 
@@ -1807,16 +1807,16 @@ def need_send_kv(self, model_input, kv_caches) -> bool:
             kv_caches: vLLM's paged memory
         """
 
+        if self.vllm_config.kv_transfer_config is None:
+            return False
+
         prefill_meta = model_input.attn_metadata.prefill_metadata
 
         # check if the current run is profiling
         is_profile_run = (kv_caches[0].numel() == 0)
         # check if the current run is prefill
         is_prefill_run = prefill_meta is not None
 
-        if self.vllm_config.kv_transfer_config is None:
-            return False
-
         return self.vllm_config.kv_transfer_config.is_kv_producer and (
             not is_profile_run) and is_prefill_run