Starstruk Logo MarkStarstruk

v0.1.12

oumi-ai/oumiv0.1.12Apr 16, 2025by jgreer013

AI Summary

Updates core dependencies (Transformers 4.51.0, TRL 0.16), adds Pixmo vision-language datasets, and improves remote inference error handling.

Key Highlights

  • Transformers 4.51.0 and TRL 0.16 updates.
  • Pixmo vision-language datasets added.
  • Remote inference improvements (GCP API URL construction).

New Features

  • Transformers and TRL version upgrades.
  • Pixmo datasets integration.
  • GRPO evaluation notebook for letter counting.
  • Remote inference error checking for `api_key`.
  • Custom processor args support in `ModelParams`.
  • BerryBench evaluation support.

Full Release Notes

## What's Changed
* Add `vllm` to `gpu` optional dependencies by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1614
* [HallOumi] Update inference notebook by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1613
* Update llama4 GCP jobs for non-dev environments. by @taenin in https://github.com/oumi-ai/oumi/pull/1621
* Update transformers to 4.51.0 by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1620
* Lazy load skypilot by @taenin in https://github.com/oumi-ai/oumi/pull/1622
* Add additional_model_kwargs and additional_trainer_kwargs to train function by @hommayushi3 in https://github.com/oumi-ai/oumi/pull/1624
* Added 3 Pixmo vision-language datasets by @jrwana in https://github.com/oumi-ai/oumi/pull/1523
* [GRPO] Add notebook to demonstrate GRPO & evaluation for letter counting by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1625
* [Remote Inference] Update Default Params by @kaisopos in https://github.com/oumi-ai/oumi/pull/1630
* Update trl to 0.16 by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1631
* Support custom `processor args` in `ModelParams` by @nikg4 in https://github.com/oumi-ai/oumi/pull/1634
* Support BerryBench evaluation by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1635
* [Remote Inference] Error checking for `api_key` by @kaisopos in https://github.com/oumi-ai/oumi/pull/1638
* Rename cnn_mnist_example to cnn_mnist_tutorial by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1640
* [Remote Inference][GCP] Constructing `api_url` from the Project ID and Region by @kaisopos in https://github.com/oumi-ai/oumi/pull/1636

## New Contributors
* @jrwana made their first contribution in https://github.com/oumi-ai/oumi/pull/1523

**Full Changelog**: https://github.com/oumi-ai/oumi/compare/v0.1.11...v0.1.12