[fix]修复test_spec_decode_worker中的错误

vllm/spec_decode/spec_decode_worker.py

@@ -550,14 +550,15 @@ class SpecDecodeWorker(LoraNotSupportedWorkerBase):
                     hidden_states, execute_model_req.seq_group_metadata_list)
                 
         # Store logits from target model execution.
-        logits = sampler_output.logits
-        if logits is not None:
-            if self.previous_logits is None:
-                self.previous_logits = Logits(
-                    logits, execute_model_req.seq_group_metadata_list)
-            else:
-                self.previous_logits.update(
-                    logits, execute_model_req.seq_group_metadata_list)
+        if self.tree_style_spec_decoding:
+            logits = sampler_output.logits
+            if logits is not None:
+                if self.previous_logits is None:
+                    self.previous_logits = Logits(
+                        logits, execute_model_req.seq_group_metadata_list)
+                else:
+                    self.previous_logits.update(
+                        logits, execute_model_req.seq_group_metadata_list)
 
         if not skip_proposer:
             # We prepare the prefill hidden states here so that there no