[BugFix]fix mtp in ucm (#321)

* fix mtp in ucm

Author: NaganooMei

ucm/integration/vllm/uc_connector.py

@@ -334,9 +334,9 @@ def wait_for_layer_load(self, layer_name: str) -> None:
         if self.layerwise_load_tasks:
             logger.debug(f"Waiting for layer {self.current_layer} to be loaded")
 
-        assert (
-            self.current_layer < self.num_layers
-        ), "The current layer should be less than total layers!"
+        if self.current_layer >= self.num_layers:
+            return
+
         for request_id, layer_to_task in self.layerwise_load_tasks.items():
             if request_id in self._load_failed_reqs:
                 continue
@@ -384,6 +384,9 @@ def save_kv_layer(
         if not self.use_layerwise:
             return
 
+        if self.current_layer > self.num_layers:
+            return
+
         metadata = self._get_connector_metadata()
         assert isinstance(metadata, UCConnectorV1Metadata)