issue/340 - fix(llm): guard PD config and restore static scheduler step

Author: ma-hang

python/infinilm/config/engine_config.py

@@ -43,3 +43,11 @@ class EngineConfig:
     attn_backend: str = "default"
     skip_load: bool = False
     kv_transfer_config: Optional[KVTransferConfig] = None
+
+    def __post_init__(self) -> None:
+        if (
+            self.kv_transfer_config is not None
+            and self.kv_transfer_config.kv_connector
+            and self.cache_type != "paged"
+        ):
+            raise ValueError("kv_transfer_config requires cache_type='paged'")

python/infinilm/llm/model_runner/model_runner.py

@@ -86,7 +86,10 @@ def __init__(self, config: EngineConfig):
 
         # Initialize KV connector
         self.kv_connector = None
-        if self.kv_transfer_config is not None:
+        if (
+            self.kv_transfer_config is not None
+            and self.kv_transfer_config.kv_connector
+        ):
             connector_name = self.kv_transfer_config.kv_connector
             self.kv_connector = KVConnectorFactory.create_connector(
                 connector_name=connector_name,

python/infinilm/llm/static_scheduler.py

@@ -223,6 +223,10 @@ def update_cache(self):
             f"update_cache: cached_block_hashes now has {len(self.cached_block_hashes)} blocks"
         )
 
+    def update_from_output(self, model_output):
+        """Static cache has no scheduler-side connector state to update."""
+        return None
+
     def complete_requests(self, requests: List[InferenceRequest]):
         """Handle completed requests."""
         for req in requests: