Skip to content

v0.9.2 - arbitrary token log-prob aggregation

Compare
Choose a tag to compare
@kddubey kddubey released this 31 Jul 01:04
· 23 commits to main since this release
b3af1e6

Breaking changes

None

New features

  • The agg_log_probs function applies func instead of np.exp ∘ func. So if you want average token log-probabilities, set func=np.mean

Bug fixes

None