fix dict type

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

Author: wangxiyuan

vllm_ascend/spec_decode/mtp_proposer.py

@@ -1,5 +1,4 @@
 import types
-from typing import Dict
 
 import torch
 import torch.nn as nn
@@ -187,7 +186,7 @@ def generate_token_ids(self,
                            hidden_states: torch.Tensor = None,
                            attn_metadata=None,
                            aux_hidden_states: torch.Tensor = None):
-        if attn_metadata is not None and isinstance(attn_metadata, Dict):
+        if attn_metadata is not None and isinstance(attn_metadata, dict):
             attn_metadata = attn_metadata['model.layers.0.self_attn.attn']
         next_token_ids: list[int] = []
         for i, token_ids in enumerate(valid_sampled_token_ids):

vllm_ascend/torchair/torchair_model_runner.py

@@ -19,7 +19,6 @@
 
 import math
 import types
-from typing import Optional, Dict
 
 import torch
 import torch.distributed as dist
@@ -135,7 +134,7 @@ def _generate_dummy_run_hidden_states(self, with_prefill,
                                           is_torchair_compile, input_ids,
                                           positions, attn_metadata, num_tokens,
                                           intermediate_tensors, inputs_embeds):
-        if attn_metadata is not None and isinstance(attn_metadata, Dict):
+        if attn_metadata is not None and isinstance(attn_metadata, dict):
             attn_metadata = attn_metadata['model.layers.0.self_attn.attn']
         if not with_prefill:
             # Only mark static while compiling
@@ -282,7 +281,7 @@ def _generate_process_reqs_hidden_states(self, attn_metadata, with_prefill,
                                              input_ids, positions,
                                              intermediate_tensors,
                                              inputs_embeds):
-        if attn_metadata is not None and isinstance(attn_metadata, Dict):
+        if attn_metadata is not None and isinstance(attn_metadata, dict):
             attn_metadata = attn_metadata['model.layers.0.self_attn.attn']
         model_kwargs = {
             "kv_caches": self.kv_caches,