At least tell the user that ngram verify is greedy! (#13039)
This commit is contained in:
@@ -419,6 +419,9 @@ class NgramVerifyInput(SpecInput):
|
|||||||
self._greedy_verify(batch, logits_output)
|
self._greedy_verify(batch, logits_output)
|
||||||
else:
|
else:
|
||||||
# NOTE: Compared with greedy_verify, the performance of _sampling_verify is relatively poor.
|
# NOTE: Compared with greedy_verify, the performance of _sampling_verify is relatively poor.
|
||||||
|
logger.warning(
|
||||||
|
"Currently Ngram speculative sampling forces to use greedy verification."
|
||||||
|
)
|
||||||
self._greedy_verify(batch, logits_output)
|
self._greedy_verify(batch, logits_output)
|
||||||
# self._sampling_verify(batch, logits_output, sampling_info)
|
# self._sampling_verify(batch, logits_output, sampling_info)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user