Adjust KL_THRESHOLD for log probability calculations (#30101)

This commit is contained in:
Cheng Wan
2026-07-03 23:53:51 -07:00
committed by GitHub
parent ad744c6c6b
commit 88db9e033a
@@ -46,7 +46,7 @@ EXPERTS_SHARED_OUTER_LORAS = True
PREFILL_ATTENTION_BACKEND = "fa4"
DECODE_ATTENTION_BACKEND = "fa4"
KL_THRESHOLD = 5e-3
KL_THRESHOLD = 6e-3 # it was 5e-3 with KL 4.766e-3. KL now is 5.008e-3.
def kl_v2(a, b):