[CI/test] Fix Eagle Correctness Test (#17209)

Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
Woosuk Kwon
2025-04-25 23:40:36 -07:00
committed by GitHub
parent b07bf83c7d
commit 513f074766
2 changed files with 5 additions and 4 deletions

View File

@@ -2220,9 +2220,10 @@ class SpeculativeConfig:
excluding anything before input ids/embeddings and after
the final hidden states.
"""
# no factors to consider.
# spec decode does not use `torch.compile` yet.
factors: list[Any] = []
# Eagle3 affects the computation graph because it returns intermediate
# hidden states in addition to the final hidden state.
factors.append(self.method == "eagle3")
hash_str = hashlib.md5(str(factors).encode(),
usedforsecurity=False).hexdigest()
return hash_str