format

Signed-off-by: shen-shanshan <467638484@qq.com>

Author: shen-shanshan

vllm_ascend/attention/attention_v1.py

@@ -361,6 +361,7 @@ def _forward_prefill_no_cache(
                                        num_heads=self.num_heads,
                                        num_kv_heads=self.num_kv_heads,
                                        out=output)
+        assert output is not None
         return output[:num_tokens, :, :]
 
     def _forward_prefill_cache_hit(