File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -148,7 +148,7 @@ def test_eagle_correctness(
148
148
print (f"ref_output: { ref_output .outputs [0 ].text } " )
149
149
print (f"spec_output: { spec_output .outputs [0 ].text } " )
150
150
151
- # Heuristic: expect at least 70 % of the prompts to match exactly
151
+ # Heuristic: expect at least 66 % of the prompts to match exactly
152
152
# Upon failure, inspect the outputs to check for inaccuracy.
153
- assert matches > int (0.7 * len (ref_outputs ))
153
+ assert matches > int (0.66 * len (ref_outputs ))
154
154
del spec_llm
Original file line number Diff line number Diff line change @@ -2220,9 +2220,10 @@ def compute_hash(self) -> str:
2220
2220
excluding anything before input ids/embeddings and after
2221
2221
the final hidden states.
2222
2222
"""
2223
- # no factors to consider.
2224
- # spec decode does not use `torch.compile` yet.
2225
2223
factors : list [Any ] = []
2224
+ # Eagle3 affects the computation graph because it returns intermediate
2225
+ # hidden states in addition to the final hidden state.
2226
+ factors .append (self .method == "eagle3" )
2226
2227
hash_str = hashlib .md5 (str (factors ).encode (),
2227
2228
usedforsecurity = False ).hexdigest ()
2228
2229
return hash_str
You can’t perform that action at this time.
0 commit comments