fix ds

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>

Author: wangxiyuan

vllm_ascend/worker/model_runner_v1.py

@@ -2436,6 +2436,7 @@ def align_memory(tensor: torch.Tensor, alignment: int) -> torch.Tensor:
             for layer_name in kv_cache_group.layer_names:
                 if layer_name in self.runner_only_attn_layers:
                     continue
+                num_blocks = None
                 if not self.model_config.is_deepseek_mla:
                     raw_tensor = kv_cache_raw_tensors[layer_name]
                     assert raw_tensor.numel(