Skip to content

v0.2.0

Latest
Compare
Choose a tag to compare
@jvamvas jvamvas released this 27 Dec 10:53
· 11 commits to main since this release

Breaking changes:

  • Rename MBRGenerationConfig to MBRConfig
  • MetricRunner now returns a MetricOutput dict instead of the raw tensor of scores.

New features:

  • Make the size of the metric cache configurable via MBRConfig.metric_cache_size
  • Allow that the number of references can be larger than the number of samples (if generated separately from the samples)

Other changes:

  • Remove GenerationConfig as parent class of MBRConfig
  • Use FIFO cache instead of LRU for metric cache
  • Extend caching to CometMetricRunner