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(
148148 print (f"ref_output: { ref_output .outputs [0 ].text } " )
149149 print (f"spec_output: { spec_output .outputs [0 ].text } " )
150150
151- # Heuristic: expect at least 70 % of the prompts to match exactly
151+ # Heuristic: expect at least 66 % of the prompts to match exactly
152152 # 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 ))
154154 del spec_llm
Original file line number Diff line number Diff line change @@ -2220,9 +2220,10 @@ def compute_hash(self) -> str:
22202220 excluding anything before input ids/embeddings and after
22212221 the final hidden states.
22222222 """
2223- # no factors to consider.
2224- # spec decode does not use `torch.compile` yet.
22252223 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" )
22262227 hash_str = hashlib .md5 (str (factors ).encode (),
22272228 usedforsecurity = False ).hexdigest ()
22282229 return hash_str
You can’t perform that action at this time.
0 commit comments