Skip to content

Keep logits in bf16 (#182)#1835

Merged
regisss merged 1 commit into
huggingface:mainfrom
HabanaAI:auto-pr-bff1365
Mar 14, 2025
Merged

Keep logits in bf16 (#182)#1835
regisss merged 1 commit into
huggingface:mainfrom
HabanaAI:auto-pr-bff1365

Conversation

@jaygala223
Copy link
Copy Markdown

Author: Jay Gala jaygala@habana.ai

What does this PR do?

Logits are converted to float32. This increases memory requirements and is not required for models running bf16

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

Author:    Jay Gala <jaygala@habana.ai>

Co-authored-by: root <root@jaygala-vm-u22.habana-labs.com>
@jaygala223 jaygala223 requested a review from regisss as a code owner March 10, 2025 08:38
@libinta libinta added run-test Run CI for PRs from external contributors transformers_future labels Mar 12, 2025
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

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.

Copy link
Copy Markdown
Collaborator

@regisss regisss left a comment

Choose a reason for hiding this comment

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

LGTM

@regisss regisss merged commit e788238 into huggingface:main Mar 14, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

run-test Run CI for PRs from external contributors transformers_future

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants