File tree 1 file changed +10
-7
lines changed 1 file changed +10
-7
lines changed Original file line number Diff line number Diff line change @@ -47,13 +47,16 @@ def log(self):
47
47
num_draft_tokens = np .sum (self .num_draft_tokens )
48
48
num_accepted_tokens = np .sum (self .num_accepted_tokens )
49
49
50
- draft_acceptance_rate = (num_accepted_tokens / num_draft_tokens
51
- if num_draft_tokens > 0 else float ("nan" ))
50
+ draft_acceptance_rate = (num_accepted_tokens / num_draft_tokens *
51
+ 100 if num_draft_tokens > 0 else float ("nan" ))
52
52
53
53
logger .info (
54
- "Speculative metrics: "
55
- "Draft acceptance rate: %.3f, "
56
- "Number of accepted tokens: %d, "
57
- "Number of draft tokens: %d, " , draft_acceptance_rate ,
58
- num_accepted_tokens , num_draft_tokens )
54
+ "SpecDecoding metrics: "
55
+ "Draft acceptance rate: %.1f%%, "
56
+ "Accepted: %d tokens, "
57
+ "Drafted: %d tokens" ,
58
+ draft_acceptance_rate ,
59
+ num_accepted_tokens ,
60
+ num_draft_tokens ,
61
+ )
59
62
self .reset ()
You can’t perform that action at this time.
0 commit comments