Skip to content

Update pegasus model card #38675

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Jun 11, 2025
Merged

Conversation

dross20
Copy link
Contributor

@dross20 dross20 commented Jun 8, 2025

What does this PR do?

This PR replaces the Pegasus model card with a new model card matching the format introduced in #36979.

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you read the contributor guideline,
    Pull Request section?
  • Was this discussed/approved via a Github issue or the forum? Please add a link
    to it if that's the case.
  • Did you make sure to update the documentation with your changes? Here are the
    documentation guidelines, and
    here are tips on formatting docstrings.
  • Did you write any new necessary tests?

Who can review?

@stevhliu

Copy link
Member

@stevhliu stevhliu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Very nice and clean update! ✨

@dross20
Copy link
Contributor Author

dross20 commented Jun 10, 2025

Looks like those errors I was running into were caused by my an issue with my local environment - sorry about that! Pegasus does in fact support sdpa and flash_attention_2, so we can leave those badges. It also supports cache_implementation="static". I've updated the code examples accordingly.

Thank you for your suggestions, and let me know if you have any more! @stevhliu

Copy link
Member

@stevhliu stevhliu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! 🎉

@stevhliu stevhliu force-pushed the update-pegasus-model-card branch from f933d58 to 4bdc74c Compare June 11, 2025 17:20
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@stevhliu stevhliu merged commit bb44d2a into huggingface:main Jun 11, 2025
10 checks passed
lmarshall12 pushed a commit to lmarshall12/transformers that referenced this pull request Jun 12, 2025
* Update Pegasus model card

* Fix transformers-cli command

* Update code examples to use bfloat16

* Reverted code examples to use float16

* Fix typo, update checkpoints link

* Update str formatting in code examples

* Apply suggestions from code review

Co-authored-by: Steven Liu <[email protected]>

* Fix typo

* Remove inaccurate badges

* Revert badge removal

* Apply suggestions from code review

Co-authored-by: Steven Liu <[email protected]>

* Include cache_implementation argument in quantization example

---------

Co-authored-by: Steven Liu <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants